From 7ba2ec7612d94a18b852d54f66819b73fa207803 Mon Sep 17 00:00:00 2001
From: "ext-ensky-jeremy.jj.he" <jeremy.jj.he@foxconn.com>
Date: Wed, 13 Aug 2025 09:40:25 +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 d773dd2..1ae26b1 100644
--- a/src/service.py
+++ b/src/service.py
@@ -1,6 +1,6 @@
 import bentoml
 import joblib
-
+import re
 my_image = bentoml.images.Image(python_version="3.10") \
     .python_packages("scikit-learn", "numpy")
 
@@ -20,6 +20,7 @@ class NewsClassifierService:
     def classify(self,text: str) -> dict:
         categories = ['Competition News','financial news','Medical news','sports news']
         cleaned_text = text.strip().replace("\n", "")
+        cleaned_text = re.sub(r'[\x00-\x1F]', '', cleaned_text)
         token_text =self.model_vectorizer.transform([cleaned_text])
         prediction = self.model_logistic.predict(token_text)
         print(f"Prediction: {prediction}")