Update app.py

app.py

@@ -84,27 +84,27 @@ if st.button("submit"):
     words=text.split()
     v=preprocess(words)
     output=model(v)
+    st.write (output.argmax())
+    # review_status = {
+    #         0: ("✅ Its a Fake news","#FF4500" ),  # Green
+    #         1: ("❌ Its not a Fake news ", "#32CD32")   # Red-Orange
+    #      }
     
-    review_status = {
-            0: ("✅ Its a Fake news","#FF4500" ),  # Green
-            1: ("❌ Its not a Fake news ", "#32CD32")   # Red-Orange
-         }
-    
-        # Get message and color based on prediction
-    message, color = review_status.get(output.argmax(), ("❓ Unknown Prediction", "#808080"))
-
-    # Display styled result
-    st.markdown(f"""
-        <div style="
-            padding: 15px; 
-            background-color: {color}; 
-            border-radius: 10px; 
-            text-align: center; 
-            font-size: 18px; 
-            font-weight: bold; 
-            color: white;">
-            {message}
-        </div>
-    """, unsafe_allow_html=True)
+    #     # Get message and color based on prediction
+    # message, color = review_status.get(output.argmax(), ("❓ Unknown Prediction", "#808080"))
+
+    # # Display styled result
+    # st.markdown(f"""
+    #     <div style="
+    #         padding: 15px; 
+    #         background-color: {color}; 
+    #         border-radius: 10px; 
+    #         text-align: center; 
+    #         font-size: 18px; 
+    #         font-weight: bold; 
+    #         color: white;">
+    #         {message}
+    #     </div>
+    # """, unsafe_allow_html=True)