Update app.py

app.py

@@ -432,11 +432,18 @@ class ForgeryDetector:
 
             region_mask = region_mask.astype(bool)
             
-            # Extract features using ORIGINAL image (not preprocessed)
-            # This matches how the classifier was trained
+            # Extract features using PREPROCESSED image (matches training)
+            # Resize region_mask to match preprocessed dimensions (384x384)
+            preprocessed_h, preprocessed_w = preprocessed.shape[:2]
+            region_mask_for_features = cv2.resize(
+                region['region_mask'].astype(np.uint8),
+                (preprocessed_w, preprocessed_h),
+                interpolation=cv2.INTER_NEAREST
+            )
+            
             features = self.feature_extractor.extract(
-                original_image / 255.0,  # Normalize to [0, 1]
-                region['region_mask'],
+                preprocessed,
+                region_mask_for_features,
                 [f.cpu() for f in decoder_features]
             )