Refactor sentiment model initialization to use AutoTokenizer and AutoModelForSequenceClassification

model/sentiment.py

@@ -1,8 +1,10 @@
-from transformers import pipeline
+from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 from typing import Any
 
 class Sentiment:
     def __init__(self, line: str) -> (list | list[Any] | Any | None):
-        self.pipe = pipeline("text-classification", model="MilaNLProc/feel-it-italian-sentiment")
+        tokenizer = AutoTokenizer.from_pretrained("MilaNLProc/feel-it-italian-sentiment")
+        model = AutoModelForSequenceClassification.from_pretrained("MilaNLProc/feel-it-italian-sentiment")
+        self.pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
         self.result = self.pipe(line)