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 |
|