nt_web_template/bentoml_rep
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix:service some bug

Browse Source
This commit is contained in:
ext-ensky-jeremy.jj.he
2025-08-13 09:31:23 +08:00
parent f2fe78cce4
commit ae2ba0893a
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/service.py
View File

@@ -19,7 +19,8 @@ class NewsClassifierService:
@bentoml.api @bentoml.api
def classify(self,text: str) -> dict: def classify(self,text: str) -> dict:
categories = ['Competition News','financial news','Medical news','sports news'] 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) prediction = self.model_logistic.predict(token_text)
print(f"Prediction: {prediction}") print(f"Prediction: {prediction}")
predict = categories[prediction[0]] predict = categories[prediction[0]]