Update app.py

app.py

@@ -6,7 +6,7 @@ The interface allows users to upload an image and receive a caption.
 import gradio as gr
 import spaces
 import torch
-from transformers import Blip2Processor, Blip2ForConditionalGeneration, BitsAndBytesConfig
+from transformers import Blip2Processor, Blip2ForConditionalGeneration
 from PIL import Image
 
 # Define device
@@ -61,14 +61,14 @@ def caption_image(image: Image.Image) -> str:
 
 # Constants for Gradio interface configuration
 IMAGE_TYPE = "pil"
-OUTPUT_TYPE = "text"
+OUTPUT_TYPE = gr.Textbox(label="Caption")
 
 # Define the Gradio interface for image captioning
 demo = gr.Interface(
     fn=caption_image,
     inputs=gr.Image(type=IMAGE_TYPE),
     outputs=OUTPUT_TYPE,
-    title="Image Captioning with BLIP",
+    title="Image Captioning with BLIP-2",
     description="Upload an image to generate a caption."
 )