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,
|