Spaces:
Runtime error
Runtime error
Rob Caamano
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,8 @@ text = st.text_area("Input text", demo, height=250)
|
|
14 |
model_options = {
|
15 |
"DistilBERT Base Uncased (SST-2)": "distilbert-base-uncased-finetuned-sst-2-english",
|
16 |
"Fine-tuned Toxicity Model": "RobCaamano/toxicity_distilbert",
|
17 |
-
"Fine-tuned Toxicity Model (RObert)": "RobCaamano/toxicity_RObert",
|
|
|
18 |
}
|
19 |
selected_model = st.selectbox("Select Model", options=list(model_options.keys()))
|
20 |
|
|
|
14 |
model_options = {
|
15 |
"DistilBERT Base Uncased (SST-2)": "distilbert-base-uncased-finetuned-sst-2-english",
|
16 |
"Fine-tuned Toxicity Model": "RobCaamano/toxicity_distilbert",
|
17 |
+
"Fine-tuned Toxicity Model (RObert)": "RobCaamano/toxicity_RObert",
|
18 |
+
"Model 3.0": "RobCaamano/toxicity_RObert2"
|
19 |
}
|
20 |
selected_model = st.selectbox("Select Model", options=list(model_options.keys()))
|
21 |
|