From ae2ba0893afb322f1853fdf9619a739f70e27af0 Mon Sep 17 00:00:00 2001
From: "ext-ensky-jeremy.jj.he" <jeremy.jj.he@foxconn.com>
Date: Wed, 13 Aug 2025 09:31:23 +0800
Subject: [PATCH] fix:service some bug

---
 src/service.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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]]