Update main.py

main.py

@@ -6,7 +6,10 @@ import database
 
 app = FastAPI()
 models.database.Base.metadata.create_all(bind=database.engine)
-classifier = pipeline("sentiment-analysis")
+distilled_student_sentiment_classifier = pipeline(
+    model="lxyuan/distilbert-base-multilingual-cased-sentiments-student", 
+    return_all_scores=True
+)
 
 # Database session dependency
 def get_db():
@@ -24,7 +27,7 @@ def create_sentiment_result(sentiment_result: schemas.SentimentResultCreate, tex
 
     try:
         text_content = text_input
-        sentiment_analysis_result = classifier(text_content)
+        sentiment_analysis_result = distilled_student_sentiment_classifier(text_content)
 
         # Create a new SentimentResult instance
         new_sentiment_result = models.SentimentResult(