Spaces:

onnx
/

ShuffleNet-v2

Runtime error

akhaliq HF Staff commited on Mar 16, 2022

Commit

865364e

1 Parent(s): 06ecef3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -41,7 +41,7 @@ def preprocess(img):
     transforms.ToTensor(),
     transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
     ])
-    img = mx.ndarray.array(img)
     img = transform_fn(img)
     img = img.expand_dims(axis=0) # batchify

     transforms.ToTensor(),
     transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
     ])
+    img = mxnet.ndarray.array(img)
     img = transform_fn(img)
     img = img.expand_dims(axis=0) # batchify