Araeynn commited on
Commit
7230f37
·
verified ·
1 Parent(s): 8dd3d1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -12,8 +12,9 @@ def image_classifier(inp):
12
  hf = os.getenv("HF")
13
  r = asyncio.run(client.zero_shot_image_classification("why.png", candidate_labels=["mouth", "teeth", "not mouth"]))
14
  c = {}
15
- c[r[0]["label"]] = r[0]["score"] + r[1]["score"]
16
- c[r[2]["label"]] = r[2]["score"]
 
17
  return c
18
 
19
 
 
12
  hf = os.getenv("HF")
13
  r = asyncio.run(client.zero_shot_image_classification("why.png", candidate_labels=["mouth", "teeth", "not mouth"]))
14
  c = {}
15
+ a = r[0]["score"] + r[1]["score"] + r[2]["score"]
16
+ c[r[0]["label"]] = (r[0]["score"] + r[1]["score"]) / a * 100
17
+ c[r[2]["label"]] = r[2]["score"] / a * 100
18
  return c
19
 
20