Update streamlit_app.py

src/streamlit_app.py

@@ -50,13 +50,13 @@ tsd_path = "Huydb/PhoBERT-toxic"
 tsd_tokenizer, tsd_model = load_tsd(tsd_path)
 
 # --- Streamlit UI ---
-# st.markdown("""
-# <style> /* CSS animation & button */
-# @keyframes bgfade {0%{background-color:white;}50%{background-color:#889ECE;}100%{background-color:white;}}
-# html, body, .reportview-container, .main {height:100%!important; margin:0; padding:0; animation:bgfade 10s ease infinite;}
-#  div.stButton>button:first-child{background-color:red!important;color:white!important;border:none;}
-# </style>
-# """, unsafe_allow_html=True)
+st.markdown("""
+<style> /* CSS animation & button */
+@keyframes bgfade {0%{background-color:white;}50%{background-color:#889ECE;}100%{background-color:white;}}
+html, body, .reportview-container, .main {height:100%!important; margin:0; padding:0; animation:bgfade 10s ease infinite;}
+ div.stButton>button:first-child{background-color:red!important;color:white!important;border:none;}
+</style>
+""", unsafe_allow_html=True)
 
 st.title("🔊🤬 Toxic Spans Detection from Audio")
 uploaded_audio = st.file_uploader("1. Upload a WAV audio file", type=["wav"])
@@ -96,4 +96,4 @@ if st.button("Transcript and Detect Toxic Spans Now"):
     labels = logits.argmax(-1)[0].cpu().tolist()
     sen_hide = highlight_toxic_span(transcript_text.split(), labels)
 
-    st.markdown(f"<h1 style='text-align: center;'>{sen_hide}</h1>", unsafe_allow_html=True)
+    st.markdown(f"<h1 style='text-align: center; color: red;'>{sen_hide}</h1>", unsafe_allow_html=True)