Rob Caamano commited on
Commit
2db0396
·
unverified ·
1 Parent(s): 528da04
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,14 +8,14 @@ from transformers import pipeline
8
 
9
  st.title("Detecting Toxic Tweets")
10
 
11
- demo = """Your words are like poison. They seep into my mind and make me feel worthless."""
12
 
13
  text = st.text_area("Input text", demo, height=250)
14
 
15
  # Add a drop-down menu for model selection
16
  model_options = {
17
  "DistilBERT Base Uncased (SST-2)": "distilbert-base-uncased-finetuned-sst-2-english",
18
- "Fine-tuned Toxicity Model": "https://huggingface.co/RobCaamano/toxicity_distilbert",
19
  }
20
  selected_model = st.selectbox("Select Model", options=list(model_options.keys()))
21
 
 
8
 
9
  st.title("Detecting Toxic Tweets")
10
 
11
+ demo = """Your words are like poison. They seep into my mind and make me feel worthless"""
12
 
13
  text = st.text_area("Input text", demo, height=250)
14
 
15
  # Add a drop-down menu for model selection
16
  model_options = {
17
  "DistilBERT Base Uncased (SST-2)": "distilbert-base-uncased-finetuned-sst-2-english",
18
+ "Fine-tuned Toxicity Model": "RobCaamano/toxicity_distilbert",
19
  }
20
  selected_model = st.selectbox("Select Model", options=list(model_options.keys()))
21