filipeclduarte/minimalFlaskAPI

1from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline2 3tokenizer = AutoTokenizer.from_pretrained("turing-usp/FinBertPTBR")4model = AutoModelForSequenceClassification.from_pretrained("turing-usp/FinBertPTBR")5 6classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)7 8def predict(data):9  return classifier(data)10  11def predict_general(data):12  classifier = pipeline("sentiment-analysis")13  return classifier(data)