Create caption.py

caption.py

@@ -0,0 +1,13 @@
+import gradio as gr
+from transformers import pipeline
+
+pipe = pipeline("image-to-text", model="salesforce/blip-image-captioning-base")
+
+def launch(input):
+    out = pipe(input)
+    return out[0]['generated_text']
+
+iface = gr.Interface(launch,
+                     inputs=gr.inputs.Image(type='pil'),
+                     outputs="text")
+iface.launch(share=True)