Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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,
|