garrettbaber commited on
Commit
d6330d3
1 Parent(s): 25c4b3c

uses anger model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-joy-intensity")
13
- tokenizer = AutoTokenizer.from_pretrained("garrettbaber/twitter-roberta-base-joy-intensity")
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