andyqin18 commited on
Commit
e294a0a
1 Parent(s): c59cf35
Files changed (1) hide show
  1. app.py +2 -0
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. \