Update app.py

app.py

@@ -332,7 +332,15 @@ class ForgeryDetector:
         binary_mask = (prob_map_resized > 0.5).astype(np.uint8)
         refined_mask = self.mask_refiner.refine(prob_map_resized, original_size=original_image.shape[:2])
         
-        # Safety check: Ensure prob_map_resized and refined_mask have same dimensions
+        # Ensure refined_mask matches prob_map_resized dimensions
+        if refined_mask.shape != prob_map_resized.shape:
+            refined_mask = cv2.resize(
+                refined_mask,
+                (prob_map_resized.shape[1], prob_map_resized.shape[0]),
+                interpolation=cv2.INTER_NEAREST
+            )
+        
+        # Safety check: Ensure prob_map_resized and refined_mask have same dimensions (fallback)
         if prob_map_resized.shape != refined_mask.shape:
             prob_map_resized = cv2.resize(
                 prob_map_resized,