Rob Caamano
commited on
Update app.py
Browse files
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 = "
|
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)
|