Update app.py

app.py

@@ -104,7 +104,7 @@ def show_mask(mask, ax, random_color=False):
         color = np.array([30/255, 144/255, 255/255, 0.6])
     h, w = mask.shape[-2:]
     mask_image = mask.reshape(h, w, 1) * color.reshape(1, 1, -1)
-    ax.imshow(mask_image, extent=(0, width, height, 0))  # Setting extent to match original image dimensions
+    ax.imshow(mask_image)
 
 ### SERVER ###
 def server(input: Inputs, output: Outputs, session: Session):