joaopdrm commited on
Commit
16bf9bb
·
1 Parent(s): aebc9b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipe
5
  class Emotionclass:
6
  def __init__(self, model: str):
7
  self.model = AutoModelForSequenceClassification.from_pretrained(model)
 
8
  self.pipeline = pipeline(
9
  "text-classification",
10
  model=self.model,
 
5
  class Emotionclass:
6
  def __init__(self, model: str):
7
  self.model = AutoModelForSequenceClassification.from_pretrained(model)
8
+ self.tokenizer = AutoTokenizer.from_pretrained(model)
9
  self.pipeline = pipeline(
10
  "text-classification",
11
  model=self.model,