comparing-captioning-models

Runtime error

nielsr HF staff commited on Jan 2, 2023

Commit

075ee8a

1 Parent(s): 27af252

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,8 +7,8 @@ torch.hub.download_url_to_file('http://images.cocodataset.org/val2017/0000000397
 git_processor = AutoProcessor.from_pretrained("microsoft/git-base-coco")
 git_model = AutoModelForCausalLM.from_pretrained("microsoft/git-base-coco")
-blip_processor = AutoProcessor.from_pretrained("Salesfoce/blip-image-captioning-base")
-blip_model = BlipForConditionalGeneration.from_pretrained("Salesfoce/blip-image-captioning-base")
 device = "cuda" if torch.cuda.is_available() else "cpu"

 git_processor = AutoProcessor.from_pretrained("microsoft/git-base-coco")
 git_model = AutoModelForCausalLM.from_pretrained("microsoft/git-base-coco")
+blip_processor = AutoProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
+blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
 device = "cuda" if torch.cuda.is_available() else "cpu"