NchourupouoM commited on
Commit
a52212d
·
verified ·
1 Parent(s): f09459b

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -2
src/streamlit_app.py CHANGED
@@ -23,8 +23,8 @@ def load_model():
23
  repo_id = "NchourupouoM/heart-disease-classifier"
24
  filename = "heart_disease_pipeline.joblib"
25
 
26
- model_path = hf_hub_download(repo_id=repo_id, filename=filename)
27
- pipeline = joblib.load(model_path)
28
  return pipeline
29
  except Exception as e:
30
  st.error(f"Erreur lors du chargement du modèle : {e}")
 
23
  repo_id = "NchourupouoM/heart-disease-classifier"
24
  filename = "heart_disease_pipeline.joblib"
25
 
26
+ # model_path = hf_hub_download(repo_id=repo_id, filename=filename)
27
+ pipeline = joblib.load("heart_disease_pipeline.joblib")
28
  return pipeline
29
  except Exception as e:
30
  st.error(f"Erreur lors du chargement du modèle : {e}")