Spaces:
Build error
Build error
sabrinabenas
commited on
Commit
β’
fb5441b
1
Parent(s):
c4387de
test
Browse files- app.py +2 -2
- model/models.py +1 -1
app.py
CHANGED
@@ -195,8 +195,8 @@ def predict_pipeline(img_input,
|
|
195 |
# pdb.set_trace()
|
196 |
#path_to_DLCmodel = DLC_models[dlc_model_input_str]
|
197 |
#pose_cfg_path = os.path.join(path_to_DLCmodel,'pose_cfg.yaml')
|
198 |
-
path_to_DLCmodel = DownloadModel(dlc_model_input_str, '
|
199 |
-
pose_cfg_path = '
|
200 |
# extract map label ids to strings
|
201 |
# pose_cfg_dict['all_joints'] is a list of one-element lists,
|
202 |
with open(pose_cfg_path, "r") as stream:
|
|
|
195 |
# pdb.set_trace()
|
196 |
#path_to_DLCmodel = DLC_models[dlc_model_input_str]
|
197 |
#pose_cfg_path = os.path.join(path_to_DLCmodel,'pose_cfg.yaml')
|
198 |
+
path_to_DLCmodel = DownloadModel(dlc_model_input_str, './model/')
|
199 |
+
pose_cfg_path = './model/pose_cfg.yaml'
|
200 |
# extract map label ids to strings
|
201 |
# pose_cfg_dict['all_joints'] is a list of one-element lists,
|
202 |
with open(pose_cfg_path, "r") as stream:
|
model/models.py
CHANGED
@@ -33,7 +33,7 @@ def DownloadModel(modelname, target_dir):
|
|
33 |
member.path = member.path[l:]
|
34 |
yield member
|
35 |
|
36 |
-
neturls = read_plainconfig("
|
37 |
|
38 |
if modelname in neturls.keys():
|
39 |
url = neturls[modelname]
|
|
|
33 |
member.path = member.path[l:]
|
34 |
yield member
|
35 |
|
36 |
+
neturls = read_plainconfig("./model/pretrained_model_urls.yaml") #FIXME
|
37 |
|
38 |
if modelname in neturls.keys():
|
39 |
url = neturls[modelname]
|