Fixing Text

tulane-cmps6730 · Apr 29, 2024 · dda8758 · dda8758
1 parent dd7f594
commit dda8758
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)