Rob Caamano commited on
Commit
8ef2341
·
unverified ·
1 Parent(s): d8d3fab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ demo = """I'm so proud of myself for accomplishing my goals today. #motivation #
11
 
12
  text = st.text_area("Input text", demo, height=250)
13
 
14
- model_name = "cardiffnlp/twitter-roberta-base-sentiment"
15
 
16
  tokenizer = AutoTokenizer.from_pretrained(model_name)
17
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
11
 
12
  text = st.text_area("Input text", demo, height=250)
13
 
14
+ model_name = "distilbert-base-uncased-finetuned-sst-2-english"
15
 
16
  tokenizer = AutoTokenizer.from_pretrained(model_name)
17
  model = AutoModelForSequenceClassification.from_pretrained(model_name)