Spaces:

nielsr
/

comparing-VQA-models

Running

nielsr HF staff commited on Jan 4, 2023

Commit

4c2e88b

•

1 Parent(s): 0799604

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,10 +13,10 @@ git_processor_large = AutoProcessor.from_pretrained("microsoft/git-large-vqav2")
 git_model_large = AutoModelForCausalLM.from_pretrained("microsoft/git-large-vqav2")
 blip_processor_base = AutoProcessor.from_pretrained("Salesforce/blip-vqa-base")
-blip_model_base = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-vqa-base")
 blip_processor_large = AutoProcessor.from_pretrained("Salesforce/blip-vqa-large")
-blip_model_large = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-vqa-large")
 vilt_processor = AutoProcessor.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
 vilt_model = ViltForQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")

 git_model_large = AutoModelForCausalLM.from_pretrained("microsoft/git-large-vqav2")
 blip_processor_base = AutoProcessor.from_pretrained("Salesforce/blip-vqa-base")
+blip_model_base = BlipForQuestionAnswering.from_pretrained("Salesforce/blip-vqa-base")
 blip_processor_large = AutoProcessor.from_pretrained("Salesforce/blip-vqa-large")
+blip_model_large = BlipForQuestionAnswering.from_pretrained("Salesforce/blip-vqa-large")
 vilt_processor = AutoProcessor.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
 vilt_model = ViltForQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")