Update app.py

app.py

@@ -156,7 +156,7 @@ For any inquiries or feedback, please contact [email protected]
         st.write("Prediction: ", prediction)
         # st.write("Prediction[0]: ", prediction[0])
         # st.write()
-        if prediction == "TRUE":
+        if prediction.get("Real") > 0.5:
             st.write("This news is predicted to be **real**.:muscle:")
         else:
             st.write("This news is predicted to be **fake**.:shit:")