Update app.py

app.py

@@ -2,26 +2,25 @@ import gradio as gr
 from transformers import pipeline
 
 # Load the image-to-text pipeline
-image_to_text_pipeline = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+image_to_text_pipelines = {
+    "Salesforce/blip-image-captioning-base": pipeline("image-to-text", model="Salesforce/blip-image-captioning-base"),
+    # Add more models if needed
+}
 
-def generate_caption(input_image):
-    # Generate caption for the input image
+def generate_caption(input_image, model_name="Salesforce/blip-image-captioning-base"):
+    # Generate caption for the input image using the selected model
+    image_to_text_pipeline = image_to_text_pipelines[model_name]
     caption = image_to_text_pipeline(input_image)[0]['generated_text']
     return caption
 
-# Additional configuration for the output component
-output_textbox = gr.outputs.Textbox(label="Generated Caption", type="auto", lines=5)
-
 # Interface for launching the model
 interface = gr.Interface(
     fn=generate_caption,
-    inputs=gr.inputs.Image(type='pil', label="Input Image"),
-    outputs=output_textbox,
+    inputs=gr.Image(type='pil', label="Input Image"),
+    outputs="text",
     title="Image Captioning Model",
     description="This model generates captions for images.",
     theme="default",
-    layout="vertical",
-    allow_screenshot=True
 )
 
 # Launch the interface