Update src/inference_utils.py

src/inference_utils.py

@@ -19,7 +19,8 @@ def compute_metrics(y_true, y_pred_proba, threshold=0.5):
 def add_predictions(df, probs):
     df['Predicted Probability'] = probs
     pred_col = f"{st.session_state.target_col} Prediction"
-    df[pred_col] = ['POSITIVE' if p > 0.5 else 'NEGATIVE' for p in probs]
+    threshold = 0.5
+    df[pred_col] = ['POSITIVE' if p >= threshold else 'NEGATIVE' for p in probs]
 
     df_with_gt = df[['Predicted Probability', pred_col]].join(st.session_state.targets_df)