akhaliq HF Staff commited on
Commit
d8b5d70
·
1 Parent(s): f735daf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -106,7 +106,7 @@ def get_model(ctx, model_path):
106
  mx.test_utils.download('https://s3.amazonaws.com/onnx-model-zoo/duc/city1.png')
107
 
108
  # read image as rgb
109
- im = cv.imread(im)[:, :, ::-1]
110
  # set output shape (same as input shape)
111
  result_shape = [im.shape[0],im.shape[1]]
112
  # set rgb mean of input image (used in mean subtraction)
 
106
  mx.test_utils.download('https://s3.amazonaws.com/onnx-model-zoo/duc/city1.png')
107
 
108
  # read image as rgb
109
+ im = cv.imread('city1.png')[:, :, ::-1]
110
  # set output shape (same as input shape)
111
  result_shape = [im.shape[0],im.shape[1]]
112
  # set rgb mean of input image (used in mean subtraction)