Update app.py

app.py

@@ -2,13 +2,15 @@ import gradio as gr
 import cv2
 
 def create_mask(image):
-   # Convert image to grayscale for easier segmentation
-   gray = cv2.cvtColor(image, 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_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
 
 with gr.Blocks() as demo:
    gr.Markdown(