Update app.py

app.py

@@ -246,18 +246,18 @@ def server(input: Inputs, output: Outputs, session: Session):
         axes[0].imshow(image)  
         axes[0].set_title("Image")
         
-        # Plot the probability map on the right
-        axes[1].imshow(prob)
-        axes[1].set_title("Probability Map")
+        # # Plot the probability map on the right
+        # axes[1].imshow(prob)
+        # axes[1].set_title("Probability Map")
         
-        # Plot the prediction image on the right
-        axes[2].imshow(prediction) 
-        axes[2].set_title("Prediction")
+        # # Plot the prediction image on the right
+        # axes[2].imshow(prediction) 
+        # axes[2].set_title("Prediction")
         
-        # Plot the predicted mask on the right
-        axes[3].imshow(image) 
-        show_mask(prediction, axes[3])
-        axes[3].set_title("Predicted Mask")
+        # # Plot the predicted mask on the right
+        # axes[3].imshow(image) 
+        # show_mask(prediction, axes[3])
+        # axes[3].set_title("Predicted Mask")
         
         # Hide axis ticks and labels
         for ax in axes: