me

Runtime error

streetyogi commited on Dec 19, 2022

Commit

270d6d5

1 Parent(s): 2bf61ba

Update inference_server.py

Files changed (1) hide show

inference_server.py CHANGED Viewed

@@ -15,7 +15,7 @@ def predict(input_text: str):
     strings.add(input_text)
     # Train a new model using all strings in the set
     model = Pipeline([
-        ('vectorizer', TfidVectorizer()),
         ('classifier', MultinomialNB())
     ])
     model.fit(list(strings), list(strings))

     strings.add(input_text)
     # Train a new model using all strings in the set
     model = Pipeline([
+        ('vectorizer', TfidfVectorizer()),
         ('classifier', MultinomialNB())
     ])
     model.fit(list(strings), list(strings))