Update app.py

app.py

@@ -114,6 +114,10 @@ class ForgeryDetector:
                 [f.cpu() for f in decoder_features]
             )
             
+            # Reshape features to 2D array (1, n_features) for classifier
+            if features.ndim == 1:
+                features = features.reshape(1, -1)
+            
             # Classify
             predictions, confidences = self.classifier.predict(features)
             forgery_type = int(predictions[0])