Update app.py

app.py

@@ -185,13 +185,12 @@ if st.button("Check News"):
     label = prediction[0]['label'].lower()
     confidence = prediction[0]['score']
 
-    # Fix classification issue
-    if confidence > 0.6 and label in ["real", "true", "positive"]:
-        st.success(f"✅ Result: This news is **REAL**. (Confidence: {confidence:.2f})")
-    elif confidence < 0.4 or label in ["fake", "false", "negative"]:
+    # Ensure proper classification
+    if label in ["fake", "false", "negative"] or confidence < 0.5:
         st.error(f"⚠️ Result: This news is **FAKE**. (Confidence: {confidence:.2f})")
     else:
-        st.warning(f"⚠️ Unable to classify with high confidence. Please verify manually. (Confidence: {confidence:.2f})")
+        st.success(f"✅ Result: This news is **REAL**. (Confidence: {confidence:.2f})")
+
 
 # ---- Deepfake Image Detection Section ----
 st.subheader("📸 Deepfake Image Detection")