image-captioning-with-longcap-safe

Sleeping

unography commited on Apr 16

Commit

c079340

•

1 Parent(s): 2df9852

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ DESCRIPTION = "# Image Captioning with LongCap"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model_id = "unography/blip-long-cap"
 processor = BlipProcessor.from_pretrained(model_id)
 model = BlipForConditionalGeneration.from_pretrained(model_id).to(device)

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model_id = "unography/blip-large-long-cap"
 processor = BlipProcessor.from_pretrained(model_id)
 model = BlipForConditionalGeneration.from_pretrained(model_id).to(device)