Ahsen Khaliq
commited on
Commit
·
88ed06b
1
Parent(s):
94a816d
Update app.py
Browse files
app.py
CHANGED
@@ -53,7 +53,7 @@ def inference(img):
|
|
53 |
|
54 |
for i in range(5):
|
55 |
labels = imagenet_labels[str(int(top5_indices[i]))]
|
56 |
-
prob =
|
57 |
result[labels] = prob
|
58 |
|
59 |
return result
|
|
|
53 |
|
54 |
for i in range(5):
|
55 |
labels = imagenet_labels[str(int(top5_indices[i]))]
|
56 |
+
prob = float(top5_prob[i])
|
57 |
result[labels] = prob
|
58 |
|
59 |
return result
|