Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ def predict(path):
|
|
52 |
{"data_0": img_batch.astype(np.float32)},
|
53 |
)
|
54 |
|
55 |
-
a = np.argsort(
|
56 |
results = {}
|
57 |
for i in a[0:5]:
|
58 |
results[labels[i]]=float(outputs[0][0][i])
|
|
|
52 |
{"data_0": img_batch.astype(np.float32)},
|
53 |
)
|
54 |
|
55 |
+
a = np.argsort(outputs[0].flatten())
|
56 |
results = {}
|
57 |
for i in a[0:5]:
|
58 |
results[labels[i]]=float(outputs[0][0][i])
|