Update app.py

app.py

@@ -134,9 +134,9 @@ def do_prediction(model_name, img):
             print(prediction.shape)
             
             '''
-            prediction_true = prediction_true * -1
-            prediction_true = prediction_true + 1
-            return "No numerical output", prediction_true * 255
+            #prediction_true = prediction_true * -1
+            #prediction_true = prediction_true + 1
+            return "No numerical output", prediction_true[:,:,0]# * 255
         
         # catch-all (we should not reach this)
         case _: