Spaces:

xxx1
/

vqa_blip_large

Runtime error

xxx1 commited on Feb 22, 2023

Commit

9ac0ee2

•

1 Parent(s): cf61b79

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,8 +8,8 @@ from transformers import BlipForQuestionAnswering, BlipProcessor
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-processor = BlipProcessor.from_pretrained("Salesforce/blip-vqa-large")
-model_vqa = BlipForQuestionAnswering.from_pretrained("Salesforce/blip-vqa-large").to(device)
 def inference_chat(input_image,input_text):
     inputs = processor(images=input_image, text=input_text,return_tensors="pt")

 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)
 def inference_chat(input_image,input_text):
     inputs = processor(images=input_image, text=input_text,return_tensors="pt")