Update app.py

app.py

@@ -253,7 +253,7 @@ def server(input: Inputs, output: Outputs, session: Session):
         
         # Plot the prediction image on the right
         prediction = prediction[:height, :width]
-        axes[2].imshow(prediction, extent=(0, width, height, 0))  # Setting extent to match original image dimensions
+        axes[2].imshow(prediction)  # Setting extent to match original image dimensions
         axes[2].set_title("Prediction")
         
         # Plot the predicted mask on the right