Update app.py

app.py

@@ -45,10 +45,10 @@ def apply_blur_effect(image, blur_type):
         predicted_mask = torch.argmax(logits, dim=1).squeeze().cpu().numpy()
 
         # Create a binary mask (1 = foreground, 0 = background)
-        foreground_mask = (predicted_mask == 11).astype(np.uint8)  # 'Person' class in Cityscapes dataset
+        foreground_mask = (predicted_mask == 11).astype(np.uint8)
 
         # Convert the mask into a PIL image for processing
-        mask_image = Image.fromarray((foreground_mask * 255).astype(np.uint8))
+        mask_image = Image.fromarray((foreground_mask * 255).astype(np.uint8)).resize(image.size)
 
         # Apply Gaussian blur to the entire image
         blurred_image = image.filter(ImageFilter.GaussianBlur(radius=15))