Update app.py

app.py

@@ -389,6 +389,16 @@ class ForgeryDetector:
         print(f"  - refined_mask: {refined_mask.shape}")
         print(f"  - prob_map_resized: {prob_map_resized.shape}")
         
+        # DEBUG: Save probability map visualization
+        prob_map_vis = (prob_map_resized * 255).astype(np.uint8)
+        prob_map_colored = cv2.applyColorMap(prob_map_vis, cv2.COLORMAP_JET)
+        print(f"[DEBUG] Probability map stats:")
+        print(f"  - Min: {prob_map_resized.min():.4f}")
+        print(f"  - Max: {prob_map_resized.max():.4f}")
+        print(f"  - Mean: {prob_map_resized.mean():.4f}")
+        print(f"  - Pixels > 0.3: {(prob_map_resized > 0.3).sum()}")
+        print(f"  - Pixels > 0.5: {(prob_map_resized > 0.5).sum()}")
+        
         # Extract regions
         regions = self.region_extractor.extract(refined_mask, prob_map_resized, original_image)