Index

tulane-cmps6730 · Apr 29, 2024 · 8bf4edb · 8bf4edb
1 parent 9353218
commit 8bf4edb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nlp/app/routes.py b/nlp/app/routes.py
@@ -43,7 +43,7 @@ def index():
             text = basic_process(input_field)
             text = cnn_process(text)
             predictions_proba = cnn.predict(text)
-            pred = (predictions_proba > 0.5).astype(int)[0]  # Assuming binary classification
+            pred = (predictions_proba > 0.5).astype(int)
             proba = predictions_proba[0]
         elif model_choice == 'bert':
             tokenizer = BertTokenizerFast.from_pretrained('prajjwal1/bert-mini')