diff --git a/src/service.py b/src/service.py
index 4f48fa3..d773dd2 100644
--- a/src/service.py
+++ b/src/service.py
@@ -19,7 +19,8 @@ class NewsClassifierService:
     @bentoml.api
     def classify(self,text: str) -> dict:
         categories = ['Competition News','financial news','Medical news','sports news']
-        token_text =self.model_vectorizer.transform([text])
+        cleaned_text = text.strip().replace("\n", "")
+        token_text =self.model_vectorizer.transform([cleaned_text])
         prediction = self.model_logistic.predict(token_text)
         print(f"Prediction: {prediction}")
         predict =  categories[prediction[0]]