Spaces:
Sleeping
Sleeping
Commit
·
bb73124
1
Parent(s):
19a4dfc
model update
Browse files- app_utils/model_utils.py +2 -2
app_utils/model_utils.py
CHANGED
|
@@ -15,7 +15,7 @@ lg.setLevel(RDLogger.CRITICAL)
|
|
| 15 |
# Load model and data
|
| 16 |
|
| 17 |
def load_model_components():
|
| 18 |
-
param_pth = '
|
| 19 |
with open(param_pth) as f:
|
| 20 |
params = yaml.load(f, Loader=yaml.FullLoader)
|
| 21 |
|
|
@@ -24,7 +24,7 @@ def load_model_components():
|
|
| 24 |
|
| 25 |
# load model
|
| 26 |
|
| 27 |
-
checkpoint_pth = "/
|
| 28 |
params['checkpoint_pth'] = checkpoint_pth
|
| 29 |
model = get_model(params['model'], params)
|
| 30 |
|
|
|
|
| 15 |
# Load model and data
|
| 16 |
|
| 17 |
def load_model_components():
|
| 18 |
+
param_pth = 'hparams.yaml'
|
| 19 |
with open(param_pth) as f:
|
| 20 |
params = yaml.load(f, Loader=yaml.FullLoader)
|
| 21 |
|
|
|
|
| 24 |
|
| 25 |
# load model
|
| 26 |
|
| 27 |
+
checkpoint_pth = "pretrained_models/flare.ckpt"
|
| 28 |
params['checkpoint_pth'] = checkpoint_pth
|
| 29 |
model = get_model(params['model'], params)
|
| 30 |
|