Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -59,7 +59,7 @@ def predict_class(image):
|
|
59 |
|
60 |
# UI Design for multy class output
|
61 |
def classify_image(image):
|
62 |
-
results = predict_class(image)
|
63 |
print("results is ...", results)
|
64 |
output = {labels.get(i): float(results[i]) for i in range(len(results))}
|
65 |
print("output is ...", output)
|
|
|
59 |
|
60 |
# UI Design for multy class output
|
61 |
def classify_image(image):
|
62 |
+
results = predict_class(image)[0]
|
63 |
print("results is ...", results)
|
64 |
output = {labels.get(i): float(results[i]) for i in range(len(results))}
|
65 |
print("output is ...", output)
|