Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,7 @@ def predict(path):
|
|
49 |
|
50 |
outputs = ort_session.run(
|
51 |
None,
|
52 |
-
{"
|
53 |
)
|
54 |
|
55 |
a = np.argsort(-outputs[0].flatten())
|
|
|
49 |
|
50 |
outputs = ort_session.run(
|
51 |
None,
|
52 |
+
{"data_0": img_batch.astype(np.float32)},
|
53 |
)
|
54 |
|
55 |
a = np.argsort(-outputs[0].flatten())
|