Spaces:
Runtime error
Runtime error
Rob Caamano
commited on
App 2.1
Browse files
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": "
|
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 |
|