Spaces:
Sleeping
Sleeping
Try1
Browse files
app.py
CHANGED
@@ -18,6 +18,8 @@ st.write("Currently it uses pre-trained models without fine-tuning.")
|
|
18 |
|
19 |
# Model hub
|
20 |
model_descrip = {
|
|
|
|
|
21 |
"distilbert-base-uncased-finetuned-sst-2-english": "This model is a fine-tune checkpoint of DistilBERT-base-uncased, fine-tuned on SST-2. \
|
22 |
Labels: POSITIVE; NEGATIVE ",
|
23 |
"cardiffnlp/twitter-roberta-base-sentiment": "This is a roBERTa-base model trained on ~58M tweets and finetuned for sentiment analysis with the TweetEval benchmark. \
|
|
|
18 |
|
19 |
# Model hub
|
20 |
model_descrip = {
|
21 |
+
"andyqin18/test-finetuned": "This is a customized BERT-base finetuned model that detects multiple toxicity for a text. \
|
22 |
+
Labels: toxic, severe_toxic, obscene, threat, insult, identity_hate",
|
23 |
"distilbert-base-uncased-finetuned-sst-2-english": "This model is a fine-tune checkpoint of DistilBERT-base-uncased, fine-tuned on SST-2. \
|
24 |
Labels: POSITIVE; NEGATIVE ",
|
25 |
"cardiffnlp/twitter-roberta-base-sentiment": "This is a roBERTa-base model trained on ~58M tweets and finetuned for sentiment analysis with the TweetEval benchmark. \
|