Subi003 commited on
Commit
949fb72
·
verified ·
1 Parent(s): 500aebb

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. main/helper.py +1 -2
  2. requirements.txt +5 -1
main/helper.py CHANGED
@@ -1,12 +1,11 @@
1
  # Helper functions for the model inference api
2
- import os
3
  import yaml
4
  import joblib
5
  from pathlib import Path
6
 
7
  # load yaml files to get model meta data.
8
  try:
9
- print(os.getcwd())
10
  with open(Path("model/registered_model_meta"), 'r') as f:
11
  model_metadata = yaml.safe_load(f)
12
  except:
 
1
  # Helper functions for the model inference api
2
+
3
  import yaml
4
  import joblib
5
  from pathlib import Path
6
 
7
  # load yaml files to get model meta data.
8
  try:
 
9
  with open(Path("model/registered_model_meta"), 'r') as f:
10
  model_metadata = yaml.safe_load(f)
11
  except:
requirements.txt CHANGED
@@ -1,4 +1,8 @@
1
  fastapi==0.116.1
2
  uvicorn==0.35.0
3
  joblib==1.5.1
4
- PyYAML==6.0.2
 
 
 
 
 
1
  fastapi==0.116.1
2
  uvicorn==0.35.0
3
  joblib==1.5.1
4
+ PyYAML==6.0.2
5
+ cloudpickle==3.1.1
6
+ numpy==2.2.6
7
+ pandas==2.3.1
8
+ scipy==1.13.1