Spaces:

onnx
/

Inception_v1

Runtime error

akhaliq HF Staff commited on Mar 10, 2022

Commit

8ec3ad9

1 Parent(s): c4c4850

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -49,7 +49,7 @@ def predict(path):
     outputs = ort_session.run(
         None,
-        {"data": img_batch.astype(np.float32)},
     )
     a = np.argsort(-outputs[0].flatten())

     outputs = ort_session.run(
         None,
+        {"data_0": img_batch.astype(np.float32)},
     )
     a = np.argsort(-outputs[0].flatten())