garrettbaber
commited on
Commit
•
d6330d3
1
Parent(s):
25c4b3c
uses anger model
Browse files
app.py
CHANGED
@@ -9,8 +9,8 @@ def formatOutput(modelOutput):
|
|
9 |
|
10 |
def processInput(input):
|
11 |
#load model and tokenizer
|
12 |
-
model = AutoModelForSequenceClassification.from_pretrained("garrettbaber/twitter-roberta-base-
|
13 |
-
tokenizer = AutoTokenizer.from_pretrained("garrettbaber/twitter-roberta-base-
|
14 |
#get tokens
|
15 |
tokens = tokenizer(input, return_tensors="pt")
|
16 |
#pass tokens to model
|
|
|
9 |
|
10 |
def processInput(input):
|
11 |
#load model and tokenizer
|
12 |
+
model = AutoModelForSequenceClassification.from_pretrained("garrettbaber/twitter-roberta-base-anger-intensity")
|
13 |
+
tokenizer = AutoTokenizer.from_pretrained("garrettbaber/twitter-roberta-base-anger-intensity")
|
14 |
#get tokens
|
15 |
tokens = tokenizer(input, return_tensors="pt")
|
16 |
#pass tokens to model
|