diff --git a/nlp/app/routes.py b/nlp/app/routes.py index dbe0d83..b3df4d7 100644 --- a/nlp/app/routes.py +++ b/nlp/app/routes.py @@ -39,7 +39,7 @@ def index(): proba = positive_proba else: prediction = "LOSS" - proba = 1 - positive_proba + proba = positive_proba elif model_choice == 'lr': @@ -55,7 +55,7 @@ def index(): proba = positive_proba else: prediction = "LOSS" - proba = 1 - positive_proba + proba = positive_proba elif model_choice == 'cnn':