blip-image-captioning-space-large

Runtime error

ybelkada commited on Dec 21, 2022

Commit

64ec3ab

•

1 Parent(s): 6a7d4e8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,8 +5,8 @@ from transformers import BlipForConditionalGeneration, BlipProcessor
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
-model_image_captioning = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base").to(device)
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
+model_image_captioning = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large").to(device)
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')