Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import torch
 from PIL import Image
 from lavis.models import load_model_and_preprocess
-import json
+
 
 # Load the Blip-Caption model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -20,13 +20,13 @@ def generate_caption(image_file):
         captions = model.generate({"image": image}, use_nucleus_sampling=True, num_captions=5)
         res=" "
         for i in captions:
-          res=res+", "+i
+          res=res+",     "+i
         return (res)
 
 # Set up the Gradio interface
 inputs = gr.inputs.Image(type="pil",label="Image")
 outputs = gr.Textbox(label="Captions")
-interface = gr.Interface(fn=generate_caption, inputs=inputs, outputs="text", title="Blip-Caption")
+interface = gr.Interface(fn=generate_caption, inputs=inputs, outputs=outputs, title="Blip-Caption")
 
 # Launch the interface
 interface.launch(share=True)