Update app.py

app.py

@@ -1,3 +1,6 @@
+
+
+from turtle import title
 import gradio as gr
 from transformers import pipeline
 import numpy as np
@@ -5,6 +8,23 @@ from PIL import Image
 
 
 pipe = pipeline("zero-shot-image-classification", model="mkaichristensen/echo-clip")
+images="dog.jpg"
+
+def shot(image, labels_text):
+    PIL_image = Image.fromarray(np.uint8(image)).convert('RGB')
+    labels = labels_text.split(",")
+    res = pipe(images=PIL_image, 
+           candidate_labels=labels,
+           hypothesis_template= "This is a photo of a {}")
+    return {dic["label"]: dic["score"] for dic in res}
+    
+iface = gr.Interface(shot, 
+                    ["image", "text"], 
+                    "label", 
+                    examples=[["dog.jpg", "dog,cat,bird"], 
+                              ["germany.jpg", "germany,belgium,colombia"],
+                              ["colombia.jpg", "germany,belgium,colombia"]],
+                    description="Add a picture and a list of labels separated by commas",
+                    title="Zero-shot Image Classification")
 
-gr.load("models/mkaichristensen/echo-clip").launch()
-print("Models loaded")
+iface.launch()