Update app.py

app.py

@@ -354,6 +354,19 @@ class ForgeryDetector:
         # Classify regions
         results = []
         for region in regions:
+                df = decoder_features[0]          # shape: [C, H, W]
+                _, fh, fw = df.shape
+
+                region_mask = region['region_mask']
+                if region_mask.shape != (fh, fw):
+                    region_mask = cv2.resize(
+                        region_mask.astype(np.uint8),
+                        (fw, fh),
+                        interpolation=cv2.INTER_NEAREST
+                    )
+
+                region_mask = region_mask.astype(bool)
+            
             # Extract features
             features = self.feature_extractor.extract(
                 preprocessed,