Update app.py

app.py

@@ -1,16 +1,8 @@
 import gradio as gr
-import cv2
+import imageio
 
 def create_mask(image):
-    image_array = gr.as_np_array(image)
-    
-    # Convert image to grayscale for easier segmentation
-    gray = cv2.cvtColor(image_array, cv2.COLOR_BGR2GRAY)
-    
-    # Apply thresholding to create a binary mask (adjust threshold value as needed)
-    _, mask = cv2.threshold(gray, 127, 255, cv2.THRESH_BINARY)
-    
-    return mask
+    image["mask"]
 
 with gr.Blocks() as demo:
    gr.Markdown(