Fix not working issues

app.py

@@ -173,19 +173,20 @@ def inference(image):
     image_tensor, orig_size = load_image(image_path, hypar) 
     mask = predict(net, image_tensor, orig_size, hypar, device)
     
-    cropped_mask = crop_to_signature(mask)
-    processed_mask = smooth_and_denoise(cropped_mask)
+    # Process the mask without cropping
+    processed_mask = smooth_and_denoise(mask)
 
-    # Convert processed mask to PIL image
+    # Convert processed mask to PIL image and resize to match original image
     pil_mask = Image.fromarray(processed_mask).convert('L')
     im_rgb = Image.open(image).convert("RGB")
+    pil_mask_resized = pil_mask.resize(im_rgb.size, Image.ANTIALIAS)
 
-    # 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)
+    im_dark = Image.new('RGB', im_rgb.size, (0, 0, 0))
+    
+    # Apply resized mask to images
+    im_rgba = im_rgb.copy()
+    im_rgba.putalpha(pil_mask_resized)
+    im_dark.putalpha(pil_mask_resized)
     
     return [im_rgba, pil_mask, im_dark]