Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ def classify_image(inp):
|
|
20 |
prediction = model.predict(inp)
|
21 |
label = dict((v,k) for k,v in labels.items())
|
22 |
predicted_class_indices=np.argmax(prediction,axis=1)
|
23 |
-
return {
|
24 |
|
25 |
|
26 |
|
|
|
20 |
prediction = model.predict(inp)
|
21 |
label = dict((v,k) for k,v in labels.items())
|
22 |
predicted_class_indices=np.argmax(prediction,axis=1)
|
23 |
+
return {labels[i]: float(predicted_class_indices[i]) for i in range(NUM_CLASSES)}
|
24 |
|
25 |
|
26 |
|