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
|