AdarshRL commited on
Commit
0af238e
·
verified ·
1 Parent(s): ea53e89

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -0
app.py CHANGED
@@ -11,7 +11,7 @@ HUGGINGFACE_MODEL_NAME= 'churn-model'
11
  HUGGINGFACE_USER_NAME = 'AdarshRL'
12
 
13
  # Download the model from the Model Hub
14
- model_path = hf_hub_download(repo_id=f"{HUGGINGFACE_USER_NAME}/{HUGGINGFACE_MODEL_NAME}", filename="best_churn_model.joblib")
15
 
16
  # Load the model
17
  model = joblib.load(model_path)
 
11
  HUGGINGFACE_USER_NAME = 'AdarshRL'
12
 
13
  # Download the model from the Model Hub
14
+ model_path = hf_hub_download(repo_id=f"{HUGGINGFACE_USER_NAME}/{HUGGINGFACE_MODEL_NAME}", filename="best_churn_model_v1.joblib")
15
 
16
  # Load the model
17
  model = joblib.load(model_path)
requirements.txt CHANGED
@@ -4,3 +4,4 @@ streamlit==1.43.2
4
  joblib==1.5.1
5
  scikit-learn==1.6.0
6
  xgboost==2.1.4
 
 
4
  joblib==1.5.1
5
  scikit-learn==1.6.0
6
  xgboost==2.1.4
7
+ mlflow==3.0.1