bug fix

app.py

@@ -3,7 +3,7 @@ import pandas as pd
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 
 # Function to load the pre-trained model
-ss = st.session_state.get(table_df=pd.DataFrame(), first_run = True)
+
 def load_finetune_model(model_name):
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSequenceClassification.from_pretrained(model_name)
@@ -104,6 +104,9 @@ for d in initial_table_data:
     initial_table_df = pd.concat([initial_table_df, pd.DataFrame(d)], ignore_index=True)
 # Load the model and perform toxicity analysis
 
+if "table" not in st.session_state:
+    st.session_state['table'] = initial_table_df
+
 if st.button("Analyze"):
     if not text:
         st.write("Please enter a text.")
@@ -128,15 +131,11 @@ if st.button("Analyze"):
                 # print("Before concatenation:")
                 # print(table_df)
                 # Concatenate the new data frame with the existing data frame
-                if ss.first_run:
-                    ss.table_df = pd.concat([pd.DataFrame(table_data), ss.table_df, initial_table_data], ignore_index=True)
-                    ss.first_run = False
-                else:
-                    ss.table_df = pd.concat([pd.DataFrame(table_data), ss.table_df], ignore_index=True)
+                st.session_state['table'] = pd.concat([pd.DataFrame(table_data), st.session_state['table']], ignore_index=True)
                 # print("After concatenation:")
                 # print(table_df)
                 # Update the table with the new data frame
-                st.table(ss.table_df)
+                st.table(st.session_state['table'])
             else:
                 st.empty()
                 sentiment_pipeline = load_model(selected_model)