Update app.py

app.py

@@ -184,14 +184,11 @@ if st.button("Check News"):
     label = prediction[0]['label'].lower()
     confidence = prediction[0]['score']
 
-    # Correct classification logic
-    if label == "fake" or confidence > 0.6:  # Adjusted confidence threshold
+    # Ensure classification even for low confidence cases
+    if label == "fake" or confidence >= 0.5:  
         st.error(f"⚠️ Result: This news is **FAKE**. (Confidence: {confidence:.2f})")
-    elif label == "real" and confidence > 0.6:
-        st.success(f"✅ Result: This news is **REAL**. (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")
 uploaded_image = st.file_uploader("Upload an Image", type=["jpg", "png", "jpeg"])