Update app.py

app.py

@@ -10,10 +10,13 @@ image_processor = AutoImageProcessor.from_pretrained("depth-anything/Depth-Anyth
 model = AutoModelForDepthEstimation.from_pretrained("depth-anything/Depth-Anything-V2-Small-hf")
 
 def apply_gaussian_blur(image, mask):
-   
+    """Applies Gaussian blur to the background based on a user-drawn mask."""
+    
+    if mask is None:
+        return image  # If no mask is provided, return the original image
+    
     # Ensure mask is grayscale and resized to match image dimensions
-    mask_pil = Image.fromarray(mask, mode='L')
-    mask_pil = mask_pil.resize(image.size)
+    mask_pil = Image.fromarray(mask).convert("L").resize(image.size)
     mask_array = np.array(mask_pil)
 
     # Create a blurred background
@@ -97,7 +100,7 @@ interface = gr.Interface(
     fn=process_image,
     inputs=[
         gr.Image(type="pil", label="Upload an Image"),
-        gr.Image(type="numpy", tool="sketch", shape=(256, 256), label="Draw Mask (Only for Gaussian Blur)"),
+        gr.Sketchpad(shape=(256, 256), label="Draw Mask (Only for Gaussian Blur)"),
         gr.Radio(["Gaussian Blur", "Lens Blur"], label="Choose Blur Effect")
     ],
     outputs=gr.Image(type="pil"),