Update app.py

app.py

@@ -332,6 +332,14 @@ 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
+        if prob_map_resized.shape != refined_mask.shape:
+            prob_map_resized = cv2.resize(
+                prob_map_resized,
+                (refined_mask.shape[1], refined_mask.shape[0]),
+                interpolation=cv2.INTER_LINEAR
+            )
+        
         # Extract regions
         regions = self.region_extractor.extract(refined_mask, prob_map_resized, original_image)