Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -131,8 +131,8 @@ best_preds = []
|
|
131 |
best_labels = []
|
132 |
for example in prediction:
|
133 |
i = np.argmax(example)
|
134 |
-
best_prediction = example
|
135 |
-
best_label = label_set
|
136 |
best_preds.append(best_prediction)
|
137 |
best_labels.append(best_label)
|
138 |
|
|
|
131 |
best_labels = []
|
132 |
for example in prediction:
|
133 |
i = np.argmax(example)
|
134 |
+
best_prediction = example[i]
|
135 |
+
best_label = label_set[i]
|
136 |
best_preds.append(best_prediction)
|
137 |
best_labels.append(best_label)
|
138 |
|