u

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
 from transformers import pipeline
 
-# ✅ Proper image captioning model
+# Lightweight image captioning model
 pipe = pipeline(
-    "image-to-text",
-    model="Salesforce/blip-image-captioning-base"
+    "image-text-to-text",
+    model="microsoft/git-base"
 )
 
 def describe_image(image):
@@ -19,9 +19,10 @@ demo = gr.Interface(
     fn=describe_image,
     inputs=gr.Image(type="pil"),
     outputs="text",
-    title="🖼️ Image Caption AI (Lightweight)",
-    description="Upload an image and get a description."
+    title="🖼️ Image Description AI (Lightweight)",
+    description="Upload an image and the AI will describe it."
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()
+