Subi003 commited on
Commit
cde95c2
·
verified ·
1 Parent(s): d2261d4

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. main/helper.py +2 -2
main/helper.py CHANGED
@@ -6,7 +6,7 @@ from pathlib import Path
6
 
7
  # load yaml files to get model meta data.
8
  try:
9
- with open(Path("model/registered_model_meta.yaml"), 'r') as f:
10
  model_metadata = yaml.safe_load(f)
11
  except:
12
  raise FileNotFoundError("Failed to load file having model metadata")
@@ -15,7 +15,7 @@ except:
15
  def load_model():
16
  """ Loads ML model from location path and returns the model. """
17
  try:
18
- with open(Path("model/python_model.pkl"), "rb") as f:
19
  model = joblib.load(f)
20
  return model
21
 
 
6
 
7
  # load yaml files to get model meta data.
8
  try:
9
+ with open(Path("app/model/registered_model_meta.yaml"), 'r') as f:
10
  model_metadata = yaml.safe_load(f)
11
  except:
12
  raise FileNotFoundError("Failed to load file having model metadata")
 
15
  def load_model():
16
  """ Loads ML model from location path and returns the model. """
17
  try:
18
+ with open(Path("app/model/python_model.pkl"), "rb") as f:
19
  model = joblib.load(f)
20
  return model
21