Update app.py

app.py

@@ -991,10 +991,15 @@ def predict_with_paper(image, paper_size, offset, offset_unit, finger_clearance=
         # Remove background
         objects_mask = remove_bg(cropped_image)
         
+        # Resize mask back to cropped image size
+        target_height = y_max - y_min
+        target_width = x_max - x_min
+        objects_mask_resized = cv2.resize(objects_mask, (target_width, target_height))
+        
         # Place back in full image space
         full_mask = np.zeros((image.shape[0], image.shape[1]), dtype=np.uint8)
-        full_mask[y_min:y_max, x_min:x_max] = objects_mask
-        
+        full_mask[y_min:y_max, x_min:x_max] = objects_mask_resized
+
         # Light filtering only - don't exclude paper area aggressively
         # Just remove small noise
         kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3))