Spaces:

jaimin
/

IMGCaption

Runtime error

jaimin commited on Feb 25, 2023

Commit

0b66865

1 Parent(s): 2208cae

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ blip_model = BlipForConditionalGeneration.from_pretrained("jaimin/Imagecap")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 git_model.to(device)
-blip_model_large.to(device)
 def generate_caption(processor, model, image, use_float_16=False):
     inputs = processor(images=image, return_tensors="pt").to(device)

 device = "cuda" if torch.cuda.is_available() else "cpu"
 git_model.to(device)
+blip_model.to(device)
 def generate_caption(processor, model, image, use_float_16=False):
     inputs = processor(images=image, return_tensors="pt").to(device)