Spaces:
Runtime error
Runtime error
Commit
•
e9fd96c
1
Parent(s):
1fe6c7c
Update inference.py
Browse files- inference.py +4 -0
inference.py
CHANGED
|
@@ -6,4 +6,8 @@ model = AutoModelForSequenceClassification.from_pretrained("turing-usp/FinBertPT
|
|
| 6 |
classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
|
| 7 |
|
| 8 |
def predict(data):
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
return classifier(data)
|
|
|
|
| 6 |
classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
|
| 7 |
|
| 8 |
def predict(data):
|
| 9 |
+
return classifier(data)
|
| 10 |
+
|
| 11 |
+
def predict_general(data):
|
| 12 |
+
classifier = pipeline("sentiment-analysis")
|
| 13 |
return classifier(data)
|