Alex Martin commited on
Commit
0b83fb6
1 Parent(s): 878ee24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ st.title("Toxic Tweets Sentiment Analysis")
11
  words = "Take that, you funking cat-dragon! You smell really bad!"
12
  text = st.text_area("Insert text for analysis below.", words)
13
 
14
- model_list = ["distilbert-base-uncased-finetuned-sst-2-english", "bert-base-cased", "openai/clip-vit-base-patch32", "emilyalsentzer/Bio_ClinicalBERT",
15
  "sentence-transformers/all-mpnet-base-v2", "facebook/bart-large-cnn", "openai/clip-vit-base-patch16", "speechbrain/spkrec-ecapa-voxceleb",
16
  "albert-base-v2"]
17
  model = st.selectbox("", model_list)
 
11
  words = "Take that, you funking cat-dragon! You smell really bad!"
12
  text = st.text_area("Insert text for analysis below.", words)
13
 
14
+ model_list = ["aim9061/fine-tuned-toxic-tweet-dilbert", "distilbert-base-uncased-finetuned-sst-2-english", "bert-base-cased", "openai/clip-vit-base-patch32", "emilyalsentzer/Bio_ClinicalBERT",
15
  "sentence-transformers/all-mpnet-base-v2", "facebook/bart-large-cnn", "openai/clip-vit-base-patch16", "speechbrain/spkrec-ecapa-voxceleb",
16
  "albert-base-v2"]
17
  model = st.selectbox("", model_list)