Update app.py

app.py

@@ -47,6 +47,7 @@ def visualize_model_output(prediction, img):
 
 
 def do_prediction(model_name, img):
+    img_org = np.copy(img)
     model = from_pretrained_keras(model_name)
 
     match model_name:
@@ -175,7 +176,7 @@ def do_prediction(model_name, img):
             '''
             #prediction_true = prediction_true * -1
             #prediction_true = prediction_true + 1
-            return "No numerical output", visualize_model_output(prediction_true,img)
+            return "No numerical output", visualize_model_output(prediction_true,img_org)
         
         # catch-all (we should not reach this)
         case _: