Update app.py

app.py

@@ -11,10 +11,10 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 processor = BlipProcessor.from_pretrained("Salesforce/blip-vqa-capfilt-large")
 model_vqa = BlipForQuestionAnswering.from_pretrained("Salesforce/blip-vqa-capfilt-large").to(device)
 
-from transformers import BlipProcessor, Blip2ForConditionalGeneration
+from transformers import BlipProcessor, BlipForConditionalGeneration
 
-cap_processor = BlipProcessor.from_pretrained("Salesforce/blip2-flan-t5-xl")
-cap_model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-flan-t5-xl")
+cap_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
+cap_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")