Oliver Li commited on
Commit
d8d71bd
·
1 Parent(s): 6e993f1
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -100,7 +100,7 @@ initial_table_data = [{'Text (portion)': ["who's speaking? \n you goddamn cocksu
100
  'Toxicity class 2': ['obscene'],
101
  'Class 2 probability': 0.16506287455558777}]
102
  for d in initial_table_data:
103
- pd.concat([table_df, pd.DataFrame(d)], ignore_index=True)
104
  # Load the model and perform toxicity analysis
105
  if st.button("Analyze"):
106
  if not text:
@@ -108,6 +108,7 @@ if st.button("Analyze"):
108
  else:
109
  with st.spinner("Analyzing toxicity..."):
110
  if selected_model == "Olivernyu/finetuned_bert_base_uncased":
 
111
  toxicity_detector = load_model(selected_model)
112
  outputs = toxicity_detector(text, top_k=2)
113
  category_names = ["toxic", "severe_toxic", "obscene", "threat", "insult", "identity_hate"]
@@ -123,7 +124,7 @@ if st.button("Analyze"):
123
  "Toxicity class 2": [results[1][0]],
124
  f"Class 2 probability": results[1][1]
125
  }
126
- table_df = pd.concat([table_df, pd.DataFrame(table_data)], ignore_index=True)
127
  st.table(table_df)
128
  else:
129
  sentiment_pipeline = load_model(selected_model)
 
100
  'Toxicity class 2': ['obscene'],
101
  'Class 2 probability': 0.16506287455558777}]
102
  for d in initial_table_data:
103
+ table_df = pd.concat([table_df, pd.DataFrame(d)], ignore_index=True)
104
  # Load the model and perform toxicity analysis
105
  if st.button("Analyze"):
106
  if not text:
 
108
  else:
109
  with st.spinner("Analyzing toxicity..."):
110
  if selected_model == "Olivernyu/finetuned_bert_base_uncased":
111
+ st.table(table_df)
112
  toxicity_detector = load_model(selected_model)
113
  outputs = toxicity_detector(text, top_k=2)
114
  category_names = ["toxic", "severe_toxic", "obscene", "threat", "insult", "identity_hate"]
 
124
  "Toxicity class 2": [results[1][0]],
125
  f"Class 2 probability": results[1][1]
126
  }
127
+ table_df = pd.concat([pd.DataFrame(table_data), table_df], ignore_index=True)
128
  st.table(table_df)
129
  else:
130
  sentiment_pipeline = load_model(selected_model)