Image-Caption

app.py

@@ -0,0 +1,29 @@
+import os
+import io
+import IPython.display
+from PIL import Image
+import base64 
+
+from transformers import pipeline
+import gradio as gr 
+
+get_completion = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+
+def captioner(image):
+    result = get_completion(image)
+    return result[0]['generated_text']
+
+gr.close_all()
+
+demo = gr.Interface(
+    fn=captioner,
+    inputs=[gr.Image(label="Upload image", type="pil")],
+    outputs=[gr.Textbox(label="Caption")],
+    title="Image Captioning with BLIP",
+    description="Caption any image using the BLIP model",
+    allow_flagging="never",
+)
+
+demo.launch()
+
+gr.close_all()

env/conda-meta/history

@@ -0,0 +1,3 @@
+==> 2023-08-24 13:54:48 <==
+# cmd: /home/ashutosh/anaconda3/bin/conda create --prefix ./env
+# conda version: 23.7.2

requirements.txt

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