Update inference.py

inference.py

@@ -1,6 +1,8 @@
-from transformers import pipeline
+from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 
-sentiment_pipeline = pipeline("sentiment-analysis")
+tokenizer = AutoTokenizer.from_pretrained("turing-usp/FinBertPTBR")
+model = AutoModelForSequenceClassification.from_pretrained("turing-usp/FinBertPTBR")
+classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
 
 def predict(data):
-  sentiment_pipeline(data)
+  classifier(data)