Update app.py

app.py

@@ -56,9 +56,9 @@ if option == "Text":
             st.text("Predicted Classes:")
             for i, label in enumerate(class_labels):
                 if i == predicted_class:
-                    st.text(f"[**{label}**]")
+                    st.markdown(f'<div style="background-color: #ffa07a; padding: 5px;"><strong>{label}</strong></div>', unsafe_allow_html=True)
                 else:
-                    st.text(label)
+                    st.markdown(f'<div>{label}</div>', unsafe_allow_html=True)
 
 elif option == "PDF":
     uploaded_file = st.file_uploader("Upload a PDF file", type=["pdf"])
@@ -82,6 +82,6 @@ elif option == "PDF":
                 st.text("Predicted Classes:")
                 for i, label in enumerate(class_labels):
                     if i == predicted_class:
-                        st.text(f"[**{label}**]")
+                        st.markdown(f'<div style="background-color: #ffa07a; padding: 5px;"><strong>{label}</strong></div>', unsafe_allow_html=True)
                     else:
-                        st.text(label)
+                        st.markdown(f'<div>{label}</div>', unsafe_allow_html=True)