Spaces:
Runtime error
Runtime error
Commit
·
e29864f
1
Parent(s):
e00393e
Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -5,4 +5,4 @@ model = AutoModelForSequenceClassification.from_pretrained("turing-usp/FinBertPT
|
|
| 5 |
classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
|
| 6 |
|
| 7 |
def predict(data):
|
| 8 |
-
classifier(data)
|
|
|
|
| 5 |
classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
|
| 6 |
|
| 7 |
def predict(data):
|
| 8 |
+
return classifier(data)
|