Update app.py

app.py

@@ -15,10 +15,10 @@ def watershed_segmentation(input_image, scribble_image):
     markers[scribble == 128] = 2
 
     # Apply watershed algorithm
-    cv2.watershed(image, markers)
+    cv2.watershed(input_image, markers)
 
     # Create a segmented mask
-    segmented_mask = np.zeros_like(image, dtype=np.uint8)
+    segmented_mask = np.zeros_like(input_image, dtype=np.uint8)
     segmented_mask[markers == 2] = [0, 0, 255]  # Red color for segmented regions
 
     return segmented_mask