image-captioning-with-longcap

Running on Zero

unography commited on May 4, 2024

Commit

b5ea83b

verified ·

1 Parent(s): 71820b7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ DESCRIPTION = "# Image Captioning with LongCap"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model_id = "unography/blip-long-cap"
 processor = AutoProcessor.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-long-cap-9ep"
 processor = AutoProcessor.from_pretrained(model_id)
 model = BlipForConditionalGeneration.from_pretrained(model_id).to(device)