Fix image sizing issue

app.py

@@ -175,12 +175,15 @@ def inference(image):
     
     cropped_mask = crop_to_signature(mask)
     processed_mask = smooth_and_denoise(cropped_mask)
-    
+
+    # Convert processed mask to PIL image
     pil_mask = Image.fromarray(processed_mask).convert('L')
     im_rgb = Image.open(image).convert("RGB")
-    im_dark = Image.new('RGB', im_rgb.size, (0, 0, 0))
-    
-    im_rgba = im_rgb.copy()
+
+    # Crop im_rgb and im_dark to match the size of pil_mask
+    im_rgba = im_rgb.crop((0, 0, pil_mask.width, pil_mask.height))
+    im_dark = Image.new('RGB', (pil_mask.width, pil_mask.height), (0, 0, 0))
+
     im_rgba.putalpha(pil_mask)
     im_dark.putalpha(pil_mask)