DOing work

tulane-cmps6730 · Apr 30, 2024 · 159d632 · 159d632
1 parent bb687ee
commit 159d632
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/nlp/app/routes.py b/nlp/app/routes.py
@@ -36,7 +36,8 @@ def index():
             else:
                 prediction = "LOSS"  
                 proba = 1 - positive_proba
-
+        prediction = None
+        proba = None
         elif model_choice == 'lr':
             model = lr
             text = process_text(input_field)
@@ -50,6 +51,8 @@ def index():
             else:
                 prediction == "LOSS"
                 proba = 1 - positive_proba
+        prediction = None
+        proba = None
         elif model_choice == 'cnn':
             # For CNN, assuming preprocessing is handled differently or is built-in
             model_path = '/Users/jackiecollopy/Downloads/project-reddit/nlp/cnn_model.h5'