gradio-application

app.py

@@ -0,0 +1,18 @@
+import gradio as gr
+from transformers import pipeline
+
+model_pipeline = pipeline("image-to-text",
+                          model = "Salesforce/blip-image-captioning-base")
+#990M model
+
+def main(input):
+  output = model_pipeline(input)  #globally defined model pipeline
+  return output[0]['generated_text'] #generated output captoion string
+
+interface_gradio = gr.Interface(
+    main,
+    inputs = gr.Image(type = 'pil'),
+    outputs = "text"
+)
+
+interface_gradio.launch()