image-captioning-with-longcap-safe

Sleeping

unography commited on May 4

Commit

5e43e87

•

1 Parent(s): 4e9e3d3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ import os
 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"

 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"