tifa-benchmark
/

promptcap-coco-vqa

visual-question-answering

image-captioning

Model card Files Files and versions

yushihu commited on Jan 25, 2023

Commit

cb72bec

·

1 Parent(s): 68336b4

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -67,7 +67,7 @@ import torch
 from promptcap import PromptCap_VQA
 # QA model support all UnifiedQA variants. e.g. "allenai/unifiedqa-v2-t5-large-1251000"
-vqa_model = PromptCap_VQA(promptcap_model="vqascore/promptcap-coco-vqa", vqa_model="allenai/unifiedqa-t5-base")
 if torch.cuda.is_available():
   vqa_model.cuda()

 from promptcap import PromptCap_VQA
 # QA model support all UnifiedQA variants. e.g. "allenai/unifiedqa-v2-t5-large-1251000"
+vqa_model = PromptCap_VQA(promptcap_model="vqascore/promptcap-coco-vqa", qa_model="allenai/unifiedqa-t5-base")
 if torch.cuda.is_available():
   vqa_model.cuda()