Update app.py

app.py

@@ -285,9 +285,9 @@ def main():
     if st.button("Predict"):
         predicted_label = pred(selected_text)[0]
         try:
-            st.write(f'<p style="color: #000000;">{predicted_label}</p>', unsafe_allow_html=True)
+            st.write(f'<p style="color: #000000;">{Labels[predicted_label]}</p>', unsafe_allow_html=True)
         except:
-            st.write(f'<p style="color: #000000;">{predicted_label}</p>', unsafe_allow_html=True)
+            st.write(f'<p style="color: #000000;">{Labels[predicted_label]}</p>', unsafe_allow_html=True)
         
 
 if __name__ == "__main__":