Update app.py

app.py

@@ -179,16 +179,19 @@ news_input = st.text_area("Enter News Text:", placeholder="Type here...")
 if st.button("Check News"):
     st.write("🔍 Processing...")
 
-    # Get AI model prediction
+    # Use AI model to classify the news
     prediction = fake_news_detector(news_input)
     label = prediction[0]['label'].lower()
     confidence = prediction[0]['score']
 
-    # Ensure classification even for low confidence cases
-    if label == "fake" or confidence >= 0.5:  
+    # Fix classification issue
+    if confidence < 0.5:
+        st.warning(f"⚠️ Unable to classify with high confidence. Please verify manually. (Confidence: {confidence:.2f})")
+    elif label in ["fake", "false", "negative"]:
         st.error(f"⚠️ Result: This news is **FAKE**. (Confidence: {confidence:.2f})")
     else:
         st.success(f"✅ Result: This news is **REAL**. (Confidence: {confidence:.2f})")
+
 # ---- Deepfake Image Detection Section ----
 st.subheader("📸 Deepfake Image Detection")
 uploaded_image = st.file_uploader("Upload an Image", type=["jpg", "png", "jpeg"])