Spaces:

noamrot
/

FuseCap-image-captioning

Runtime error

noamrot commited on Feb 28

Commit

3e9afa5

•

1 Parent(s): 226462f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,8 +5,8 @@ from transformers import BlipProcessor, BlipForConditionalGeneration
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-processor = BlipProcessor.from_pretrained("noamrot/FuseCap")
-model = BlipForConditionalGeneration.from_pretrained("noamrot/FuseCap").to(device)
 def inference(raw_image):
     text = "a picture of "

 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+processor = BlipProcessor.from_pretrained("noamrot/FuseCap_Image_Captioning")
+model = BlipForConditionalGeneration.from_pretrained("noamrot/FuseCap_Image_Captioning").to(device)
 def inference(raw_image):
     text = "a picture of "