NYUAD-ComNets commited on
Commit
c4b3ef8
·
verified ·
1 Parent(s): 4411f87

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -26,7 +26,7 @@ import torch
26
  from transformers import PaliGemmaProcessor, PaliGemmaForConditionalGeneration, BitsAndBytesConfig, TrainingArguments, Trainer
27
  import time
28
 
29
- model = PaliGemmaForConditionalGeneration.from_pretrained('NYUAD-ComNets/paligemma_Malaysian_plate_recognition',torch_dtype=torch.bfloat16)
30
 
31
  input_text ="extract the text from the image"
32
 
 
26
  from transformers import PaliGemmaProcessor, PaliGemmaForConditionalGeneration, BitsAndBytesConfig, TrainingArguments, Trainer
27
  import time
28
 
29
+ model = PaliGemmaForConditionalGeneration.from_pretrained('NYUAD-ComNets/VehiclePaliGemma',torch_dtype=torch.bfloat16)
30
 
31
  input_text ="extract the text from the image"
32