Update README.md

README.md

@@ -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)
+  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)