Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files- 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 |
|