zsic

app.py

@@ -14,7 +14,8 @@ def shot(image, labels_text):
     labels = labels_text.split(";;")
     res = pipe(images=PIL_image, 
            candidate_labels=labels,
-           hypothesis_template="This is a photo of {}")
+           hypothesis_template="This is a photo of {}",
+		   multi_label=True)
     return {dic["label"]: dic["score"] for dic in res}
     
 iface = gr.Interface(shot, 
@@ -22,7 +23,7 @@ iface = gr.Interface(shot,
                     "label", 
                     examples=[
 							  ["examples/test1/1.jpg", "ralph lauren;;apparel store;;ralph lauren store;;shirts;;wardrobe;;white flower"], 
-                              ["examples/test1/2.JPG", "adidas;;apparel store;;adidas store;;shirts;;wardrobe;;women training shoes;;shoes"],
+                              ["examples/test1/2.JPG", "adidas;;apparel store;;adidas store;;shirts;;wardrobe;;women training;;shoes"],
 							  
                               #["examples/test2/1.jpg", test2_labels],
                               #["examples/test2/2.jpg", test2_labels],