Update app.py
Browse files
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', '
|
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":
|