modified table display

app.py

@@ -102,13 +102,14 @@ initial_table_data = [{'Text (portion)': ["who's speaking? \n you goddamn cocksu
 for d in initial_table_data:
     table_df = pd.concat([table_df, pd.DataFrame(d)], ignore_index=True)
 # Load the model and perform toxicity analysis
+st.table(table_df)
 if st.button("Analyze"):
     if not text:
         st.write("Please enter a text.")
     else:
         with st.spinner("Analyzing toxicity..."):
             if selected_model == "Olivernyu/finetuned_bert_base_uncased":
-                st.table(table_df)
+                st.empty()
                 toxicity_detector = load_model(selected_model)
                 outputs = toxicity_detector(text, top_k=2)
                 category_names = ["toxic", "severe_toxic", "obscene", "threat", "insult", "identity_hate"]
@@ -127,6 +128,7 @@ if st.button("Analyze"):
                 table_df = pd.concat([pd.DataFrame(table_data), table_df], ignore_index=True)
                 st.table(table_df)
             else:
+                st.empty()
                 sentiment_pipeline = load_model(selected_model)
                 result = sentiment_pipeline(text)
                 st.write(f"Sentiment: {result[0]['label']} (confidence: {result[0]['score']:.2f})")