joaopdrm commited on
Commit
093fcf9
1 Parent(s): 16bf9bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -9,6 +9,7 @@ class Emotionclass:
9
  self.pipeline = pipeline(
10
  "text-classification",
11
  model=self.model,
 
12
  return_all_scores=True,
13
  )
14
 
 
9
  self.pipeline = pipeline(
10
  "text-classification",
11
  model=self.model,
12
+ tokenizer=self.tokenizer,
13
  return_all_scores=True,
14
  )
15