Update predict_model.py

predict_model.py

@@ -16,5 +16,5 @@ def run(in_image):
     
     # cv2.imwrite(f'colored-{im_path}', cv2.resize(predicted * 255., (height, width)))
     # plot_inference_images(in_image,predicted)
-
-    return cv2.resize((predicted * 255.).astype(int), (width, height))
+    out_img = cv2.resize(predicted * 255., (width, height))
+    return out_img.astype(int)