Spaces:

xMcLovinx
/

image-captioning

Sleeping

xMcLovinx commited on Mar 25, 2024

Commit

1c1de54

verified ·

1 Parent(s): 2e69fd0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base
 model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
 def execute(image):
-    inputs = processor(raw_image, return_tensors="pt")
     out = model.generate(**inputs)
     output = processor.decode(out[0], skip_special_tokens=True)
     return output

 model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
 def execute(image):
+    inputs = processor(image, return_tensors="pt")
     out = model.generate(**inputs)
     output = processor.decode(out[0], skip_special_tokens=True)
     return output