Update app.py

app.py

@@ -76,6 +76,8 @@ def predict_news(text, tokenizer, text_model):
     with torch.no_grad():
         outputs = text_model(**inputs)
     probs = torch.nn.functional.softmax(outputs.logits, dim=1)
+     # ADD THIS to see raw scores
+    st.write(f"Debug → Label 0 score: {probs[0][0]:.2f}, Label 1 score: {probs[0][1]:.2f}")
     prediction = torch.argmax(probs).item()
     confidence = torch.max(probs).item() * 100
     return prediction, confidence