Update app.py

app.py

@@ -1,16 +1,17 @@
 import gradio as gr
 from transformers import pipeline
 
-def do_action(text):
-    classifier = pipeline("zero-shot-classification",
-                          model="facebook/bart-large-mnli")
+# Rename 'text' to 'image'
+def do_action(image):
+    # Use the blip model
+    pipe = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
 
-    candidate_labels = ['gun control', 'abortion', 'gender transition', 'freedom of speech', 'immigration', 'taxes']
-    result = classifier(text, candidate_labels)
-
-    result = dict(zip(result['labels'], result['scores']))
+    result = pipe(image)
 
     return result
 
-iface = gr.Interface(fn=do_action, inputs="text", outputs="label")
-iface.launch()
+# input image, output text
+# can't use inputs="image"
+# instead use inputs=gr.Image(type='pil')
+iface = gr.Interface(fn=do_action, inputs=gr.Image(type='pil'), outputs="text")
+iface.launch()