Update app.py

app.py

@@ -10,11 +10,11 @@ def image_classifier(inp):
     inp.save("why.png") 
     sunflower_path = "why.png"
     hf = os.getenv("HF")
-    r = asyncio.run(client.zero_shot_image_classification("why.png", candidate_labels=["mouth", "teeth", "not mouth"]))
+    r = asyncio.run(client.zero_shot_image_classification("why.png", candidate_labels=["mouth or teeth", "not mouth"]))
     c = {}
-    a = r[0]["score"] + r[1]["score"] + r[2]["score"]
-    c[r[0]["label"]] = (r[0]["score"] + r[1]["score"]) / a
-    c[r[2]["label"]] = r[2]["score"] / a
+    a = r[0]["score"] + r[1]["score"]
+    c[r[0]["label"]] = r[0]["score"] / a
+    c[r[1]["label"]] = r[1]["score"] / a
     return c