Trying Salesforce/blip2-opt-6.7b-coco

app.py

@@ -8,10 +8,10 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 
 # Load the BLIP-2 model and processor
-processor = AutoProcessor.from_pretrained("Salesforce/blip2-flan-t5-xxl")
+processor = AutoProcessor.from_pretrained("Salesforce/blip2-opt-6.7b-coco")
 # Load model in int8 using bitsandbytes, and pass device_map='auto'
 model = Blip2ForConditionalGeneration.from_pretrained(
-    "Salesforce/blip2-flan-t5-xxl", load_in_8bit=True, device_map='auto'
+    "Salesforce/blip2-opt-6.7b-coco", load_in_8bit=True, device_map='auto'
 )
 
 # Uncomment lines 20, 22, 23, & 24 to begin using blip2-oopt-2.7b model