File size: 312 Bytes
cb35e39
acbd829
cf06d78
690d96a
cf06d78
acbd829
 
e29864f
1
2
3
4
5
6
7
8
from transformers import AutoModel, AutoTokenizer, pipeline

tokenizer = AutoTokenizer.from_pretrained("turing-usp/FinBertPTBR")
model = AutoModel.from_pretrained("turing-usp/FinBertPTBR")
classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)

def predict(data):
  return classifier(data)