Spaces:

darshan8950
/

image-captioning-using-blip

Sleeping

darshan8950 commited on Apr 28, 2023

Commit

90a0d85

1 Parent(s): 330716d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ def generate_caption(image_file):
         # Generate captions using the Blip-Caption model
         captions = model.generate({"image": image}, use_nucleus_sampling=True, num_captions=5)
-        return captions
 # Set up the Gradio interface
 inputs = gr.inputs.Image(type="pil",label="Image")

         # Generate captions using the Blip-Caption model
         captions = model.generate({"image": image}, use_nucleus_sampling=True, num_captions=5)
+        return captions[0]
 # Set up the Gradio interface
 inputs = gr.inputs.Image(type="pil",label="Image")