image-captioning-with-longcap-safe

Sleeping

unography commited on May 4

Commit

4e9e3d3

•

1 Parent(s): dd8705d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

 DESCRIPTION = "# Image Captioning with LongCap"
+device = torch.device("cuda" if torch.cuda.is_available() else "CPU")
+print("Using device: ", device)
+model_id = "unography/blip-long-cap"
 processor = AutoProcessor.from_pretrained(model_id)
 model = BlipForConditionalGeneration.from_pretrained(model_id).to(device)