Update app.py

app.py

@@ -173,6 +173,8 @@ def detect_deepfake_image(image_path):
 #         else:
 #             st.success(f"✅ Result: This news is **REAL**. (Confidence: {confidence:.2f})")
 # ---- Fake News Detection Section ----
+import streamlit as st
+
 st.subheader("📝 Fake News Detection")
 news_input = st.text_area("Enter News Text:", placeholder="Type here...")
 
@@ -184,8 +186,8 @@ if st.button("Check News"):
     label = prediction[0]['label'].lower()
     confidence = prediction[0]['score']
 
-    # Fix classification issue
-    if label in ["fake", "false", "negative"]:
+    # Ensure proper classification
+    if label in ["fake", "false", "negative"] or confidence < 0.7:
         st.error(f"⚠️ Result: This news is **FAKE**. (Confidence: {confidence:.2f})")
     else:
         st.success(f"✅ Result: This news is **REAL**. (Confidence: {confidence:.2f})")