Rob Caamano commited on
Commit
69dabd8
·
unverified ·
1 Parent(s): d5a39db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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", # Added new model option
 
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