app.py is updated

app.py

@@ -1,16 +1,28 @@
 import gradio as gr
 from transformers import pipeline
 
-pipe = pipeline("image-to-text",
-                model="Salesforce/blip-image-captioning-base")
+# Load the image-to-text pipeline
+image_to_text_pipeline = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
 
+def generate_caption(input_image):
+    # Generate caption for the input image
+    caption = image_to_text_pipeline(input_image)[0]['generated_text']
+    return caption
 
-def launch(input):
-    out = pipe(input)
-    return out[0]['generated_text']
+# Additional configuration for the output component
+output_textbox = gr.outputs.Textbox(label="Generated Caption", type="auto", lines=5)
 
-iface = gr.Interface(launch,
-                     inputs=gr.Image(type='pil'),
-                     outputs="text")
+# Interface for launching the model
+interface = gr.Interface(
+    fn=generate_caption,
+    inputs=gr.inputs.Image(type='pil', label="Input Image"),
+    outputs=output_textbox,
+    title="Image Captioning Model",
+    description="This model generates captions for images.",
+    theme="default",
+    layout="vertical",
+    allow_screenshot=True
+)
 
-iface.launch()
+# Launch the interface
+interface.launch()