kkatiz
/

THAI-BLIP-2

visual-question-answering

Model card Files Files and versions

kkatiz commited on May 5, 2024

Commit

ed40e84

·

verified ·

1 Parent(s): 07c57a1

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -18,8 +18,8 @@ base_model: Salesforce/blip2-opt-2.7b-coco
   device = "cuda" if torch.cuda.is_available() else "cpu"
-  processor = Blip2Processor.from_pretrained("kkatiz/THAI-BLIP-2-final_model")
-  model = Blip2ForConditionalGeneration.from_pretrained("kkatiz/THAI-BLIP-2-final_model", device_map=device, torch_dtype=torch.bfloat16)
   img = Image.open("Your image...")
   inputs = processor(images=img, return_tensors="pt").to(device, torch.bfloat16)

   device = "cuda" if torch.cuda.is_available() else "cpu"
+  processor = Blip2Processor.from_pretrained("kkatiz/THAI-BLIP-2")
+  model = Blip2ForConditionalGeneration.from_pretrained("kkatiz/THAI-BLIP-2", device_map=device, torch_dtype=torch.bfloat16)
   img = Image.open("Your image...")
   inputs = processor(images=img, return_tensors="pt").to(device, torch.bfloat16)