Update README.md

README.md

@@ -6,7 +6,7 @@ base_model:
 - ViniMig/stroke-risk-package
 pipeline_tag: tabular-classification
 ---
-# Model Card for Model ID
+# Model Card for stroke-risk-package
 
 ## Model Details
 
@@ -57,6 +57,10 @@ For preprocessing this relied on a first EDA step and a previous version of the
 - Imputation
 - Binning continuous features
 - OrdinalEncoding
-- over sampling with SMOTE on train set.
+- Over sampling with SMOTE on train set.
 
 [Preprocessing code](https://gitlab.com/ViniMig/ml_full_project/-/blob/master/src/etl_pipeline/transform_load.py?ref_type=heads)
+
+Some artifacts are saved with joblib, one of which is the custom function to apply pandas cut method for generating the bins. This joblib file is currently appearing as unsafe on HF space, which is probably expected, but all the code can be seen in the linked project repository above.
+
+This artifact is saved and stored here to be used in the streamlit app which will handle user input and preprocessing for inference.