Spaces:
Sleeping
Sleeping
Commit
·
e5ecda2
1
Parent(s):
7f5ab8a
Update app.py
Browse files
app.py
CHANGED
@@ -57,10 +57,11 @@ def predict_class(image):
|
|
57 |
def classify_image(image):
|
58 |
results = predict_class(image)
|
59 |
output = {}
|
60 |
-
for
|
61 |
-
predicted_label = labels.get(
|
62 |
-
score =
|
63 |
output[predicted_label] = score
|
|
|
64 |
return output
|
65 |
|
66 |
|
|
|
57 |
def classify_image(image):
|
58 |
results = predict_class(image)
|
59 |
output = {}
|
60 |
+
for index in range(len(results)):
|
61 |
+
predicted_label = labels.get(index)
|
62 |
+
score = results[index]
|
63 |
output[predicted_label] = score
|
64 |
+
print(output)
|
65 |
return output
|
66 |
|
67 |
|