CodyJiang commited on
Commit
618a9e4
·
1 Parent(s): 0a418b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ st.title("Toxicity Classification App")
19
 
20
  user_input = st.text_input("Input texts to analyze", "Cody Jiang is a fantastic student in CS-UY-4613!")
21
 
22
- model_names = ['distilbert-base-uncased-finetuned-sst-2-english', 'bert-base-uncased', 'roberta-base', 'Codys-Finetuning-Language-Model']
23
  model_name = st.selectbox("Select a pretrained model", model_names)
24
 
25
  if model_name == "Codys-Finetuning-Language-Model":
 
19
 
20
  user_input = st.text_input("Input texts to analyze", "Cody Jiang is a fantastic student in CS-UY-4613!")
21
 
22
+ model_names = ['distilbert-base-uncased-finetuned-sst-2-english', 'Codys-Finetuning-Language-Model']
23
  model_name = st.selectbox("Select a pretrained model", model_names)
24
 
25
  if model_name == "Codys-Finetuning-Language-Model":