Update app.py

app.py

@@ -39,9 +39,11 @@ def classify_image(img: Image.Image):
     with torch.no_grad():
         outputs = model(img_tensor)
         _, predicted = torch.max(outputs, 1)
-        label = 'Fake' if predicted.item() == 0 else 'Real'
+        print("🔍 Prediction index:", predicted.item())  # Optional debug
+        label = 'Real' if predicted.item() == 0 else 'Fake'
     return label
 
+
 # ✅ Then wrap it in predict()
 def predict(img: Image.Image):
     return classify_image(img)