Update app.py

app.py

@@ -900,8 +900,8 @@ def main():
                                 pred_class = torch.argmax(probabilities).item()
                                 confidence = probabilities[pred_class].item()
                                 
-                                # Explicit class mapping - adjust if needed based on your model
-                                pred_label = "Fake" if pred_class == 0 else "Real"
+                                # Explicit class mapping - 0 = Real, 1 = Fake
+                                pred_label = "Real" if pred_class == 0 else "Fake"
                         except Exception as e:
                             st.error(f"Error in model inference: {str(e)}")
                             import traceback