Spaces:

onnx
/

ShuffleNet-v2

Runtime error

akhaliq HF Staff commited on Mar 16, 2022

Commit

06ecef3

1 Parent(s): abda02f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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