From dda8758d5f0b8b5c2529a0104be65642b5261f23 Mon Sep 17 00:00:00 2001 From: jcollopy-tulane Date: Sun, 28 Apr 2024 21:03:00 -0500 Subject: [PATCH] Fixing Text --- nlp/app/routes.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nlp/app/routes.py b/nlp/app/routes.py index a4ceb26..62caaba 100644 --- a/nlp/app/routes.py +++ b/nlp/app/routes.py @@ -22,7 +22,8 @@ def index(): model_choice = form.model_choice.data if model_choice == 'bnb': - model = bnbtext = process_text(input_field) + model = bnb + text = process_text(input_field) text = bnb_vectorizer.transform([text]) pred_labels = bnb.predict(text) probas = bnb.predict_proba(text)