Spaces:

jaimin
/

IMGCaption

Runtime error

jaimin commited on Feb 25, 2023

Commit

2208cae

1 Parent(s): 0e48cd7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ blip_processor = AutoProcessor.from_pretrained("jaimin/Imagecap")
 blip_model = BlipForConditionalGeneration.from_pretrained("jaimin/Imagecap")
 device = "cuda" if torch.cuda.is_available() else "cpu"
-git_model_large_textcaps.to(device)
 blip_model_large.to(device)
 def generate_caption(processor, model, image, use_float_16=False):

 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):