NYUAD-ComNets commited on
Commit
28f460c
·
verified ·
1 Parent(s): 43ff9af

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -23,7 +23,7 @@ import torch
23
  from transformers import PaliGemmaProcessor, PaliGemmaForConditionalGeneration, BitsAndBytesConfig, TrainingArguments, Trainer
24
 
25
 
26
- model = PaliGemmaForConditionalGeneration.from_pretrained('NYUAD-ComNets/paligemma_age',torch_dtype=torch.bfloat16)
27
 
28
  input_text = "what is age group of person in the image?"
29
 
 
23
  from transformers import PaliGemmaProcessor, PaliGemmaForConditionalGeneration, BitsAndBytesConfig, TrainingArguments, Trainer
24
 
25
 
26
+ model = PaliGemmaForConditionalGeneration.from_pretrained('NYUAD-ComNets/ScanFacePaliGemma_Age',torch_dtype=torch.bfloat16)
27
 
28
  input_text = "what is age group of person in the image?"
29