update code

app.py

@@ -20,7 +20,7 @@ depth_model = AutoModelForDepthEstimation.from_pretrained("depth-anything/Depth-
 
 def apply_blur(image, blur_type, blur_strength):
     # Convert image to RGB
-    img = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+    img = cv2.cvtColor(image)
     
     if blur_type == "Gaussian":
         # Use Segformer for Gaussian blur
@@ -31,6 +31,7 @@ def apply_blur(image, blur_type, blur_strength):
         
         mask = logits[0, 12, :, :].detach().cpu().numpy() > -4
         mask = cv2.resize(mask.astype(np.uint8), (img.shape[1], img.shape[0]))
+        mask = cv2.bitwise_not(mask)
         
     elif blur_type == "Lens":
         # Use Depth-Anything for lens blur
@@ -51,8 +52,6 @@ def apply_blur(image, blur_type, blur_strength):
         mask = prediction[0, 0, :, :].detach().cpu().numpy() < 3
         mask = mask.astype(np.uint8)
     
-    # Invert mask using cv2
-    mask = cv2.bitwise_not(mask)
     mask = np.repeat(mask[:, :, np.newaxis], 3, axis=2)
 
     # Apply blur based on selected type