Oliver Li commited on
Commit
34b7701
·
1 Parent(s): f88671d
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -108,7 +108,6 @@ if st.button("Analyze"):
108
  else:
109
  with st.spinner("Analyzing toxicity..."):
110
  if selected_model == "Olivernyu/finetuned_bert_base_uncased":
111
- st.empty()
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"]
 
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"]