zsic

app.py

@@ -14,8 +14,7 @@ def shot(image, labels_text):
     labels = labels_text.split(";;")
     res = pipe(images=PIL_image, 
            candidate_labels=labels,
-           hypothesis_template="This is a photo of {}",
-		   multi_label=True)
+           hypothesis_template="This is a photo of {}")
     return {dic["label"]: dic["score"] for dic in res}
     
 iface = gr.Interface(shot, 
@@ -23,7 +22,9 @@ 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"],
+                              ["examples/test1/2.jpg", "adidas;;apparel store;;adidas store;;shirts;;wardrobe;;women training;;shoes"], 
+                              ["examples/test1/3.jpg", "project x;;sweet monster;;bags store;;glass windows;;hanging lights"], 
+                              ["examples/test1/4.jpg", "multi brand store;;multi brand shoe store;;shoe store;;mannequins;;adidas store;;reebok store;;puma store"],
 							  
                               #["examples/test2/1.jpg", test2_labels],
                               #["examples/test2/2.jpg", test2_labels],