Update app.py

app.py

@@ -271,12 +271,14 @@ news_input = st.text_area("Enter News Text:", placeholder="Type here...")
 
 if st.button("Check News"):
     st.write("\U0001F50D Processing...")
-    labels = ["fake news", "real news"]
-    prediction = fake_news_detector(news_input, labels)
-    label = prediction['labels'][0]
-    confidence = prediction['scores'][0]
-    
-    if label == "fake news":
+
+    labels = ["FAKE", "REAL"]  # Corrected labels
+    prediction = fake_news_detector(news_input, candidate_labels=labels)
+
+    label = prediction["labels"][0]  # Highest confidence label
+    confidence = prediction["scores"][0]  # Confidence of highest label
+
+    if label == "FAKE":
         st.error(f"⚠️ Result: This news is FAKE. (Confidence: {confidence:.2f})")
     else:
         st.success(f"✅ Result: This news is REAL. (Confidence: {confidence:.2f})")