Ahsen Khaliq
commited on
Commit
·
94a816d
1
Parent(s):
dba7cbf
Update app.py
Browse files
app.py
CHANGED
@@ -54,7 +54,7 @@ def inference(img):
|
|
54 |
for i in range(5):
|
55 |
labels = imagenet_labels[str(int(top5_indices[i]))]
|
56 |
prob = "{:.2f}%".format(float(top5_prob[i])*100)
|
57 |
-
|
58 |
|
59 |
return result
|
60 |
|
|
|
54 |
for i in range(5):
|
55 |
labels = imagenet_labels[str(int(top5_indices[i]))]
|
56 |
prob = "{:.2f}%".format(float(top5_prob[i])*100)
|
57 |
+
result[labels] = prob
|
58 |
|
59 |
return result
|
60 |
|