Upload 2 files

app.py.txt

@@ -0,0 +1,21 @@
+import gradio as gr
+from transformers import pipeline
+
+captioner = pipeline(
+    "image-to-text",
+    model="Salesforce/blip-image-captioning-base"
+)
+
+def generate_caption(image):
+    result = captioner(image)
+    return result[0]["generated_text"]
+
+demo = gr.Interface(
+    fn=generate_caption,
+    inputs=gr.Image(type="pil", label="Upload an Image"),
+    outputs=gr.Textbox(label="Generated Caption"),
+    title="AI Image Captioning App",
+    description="Upload an image and generate a caption using a Hugging Face vision-language model."
+)
+
+demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+transformers
+torch