Update app.py

app.py

@@ -229,7 +229,7 @@ def do_prediction(model_name, img):
                 img_border[:, :] = 1
                 img_border = np.repeat(img_border[:, :, np.newaxis], 3, axis=2)
                 
-            return "No numerical output", visualize_model_output(img_border,img, model_name)
+            return "No numerical output", visualize_model_output(img_border,img_org, model_name)
                 
                             
         # bitmap output