Ahsen Khaliq
commited on
Commit
·
aaae048
1
Parent(s):
dd5f104
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ def inference(img):
|
|
30 |
result = {}
|
31 |
|
32 |
for i in range(5):
|
33 |
-
result[pred_names[i][1]] =
|
34 |
|
35 |
return result
|
36 |
|
|
|
30 |
result = {}
|
31 |
|
32 |
for i in range(5):
|
33 |
+
result[pred_names[i][1]] = pred_names[i][2].item()
|
34 |
|
35 |
return result
|
36 |
|