Update main.py

main.py

@@ -49,17 +49,16 @@ def preprocess_text(text):
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     # Load the model from HuggingFace transformers library
- #   from transformers import pipeline
-    
-#    sentiment_task = pipeline("sentiment-analysis", model="cardiffnlp/twitter-roberta-base-sentiment-latest", tokenizer="cardiffnlp/twitter-roberta-base-sentiment-latest")
+     from transformers import pipeline
+     global sentiment_task 
+     sentiment_task = pipeline("text-classification", model="lxyuan/distilbert-base-multilingual-cased-sentiments-student", tokenizer= "lxyuan/distilbert-base-multilingual-cased-sentiments-student")
     # Use a pipeline as a high-level helper
-    from transformers import pipeline
-    global sentiment_task
-    pipe = pipeline("text-classification", model="SamLowe/roberta-base-go_emotions", tokenizer="SamLowe/roberta-base-go_emotions")
+ #   from transformers import pipeline
     
-    yield
+ #   pipe = pipeline("text-classification", model="SamLowe/roberta-base-go_emotions", tokenizer="SamLowe/roberta-base-go_emotions")
+     yield
     # Clean up the model and release the resources
-    del sentiment_task
+     del sentiment_task
 
 description = """
 ## Text Classification API