Update main.py

main.py

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