Update app.py

app.py

@@ -105,6 +105,12 @@ def get_model(ctx, model_path):
 # Download test image
 mx.test_utils.download('https://s3.amazonaws.com/onnx-model-zoo/duc/city1.png')
 
+# read image as rgb
+im = cv.imread(im)[:, :, ::-1]
+# set output shape (same as input shape)
+result_shape = [im.shape[0],im.shape[1]]
+# set rgb mean of input image (used in mean subtraction)
+rgb_mean = cv.mean(im)
 
 # Download ONNX model
 mx.test_utils.download('https://s3.amazonaws.com/onnx-model-zoo/duc/ResNet101_DUC_HDC.onnx')