Spaces:
Build error
Build error
sabrinabenas
commited on
Commit
β’
668cb3e
1
Parent(s):
38d9537
test download
Browse files
app.py
CHANGED
@@ -193,10 +193,10 @@ def predict_pipeline(img_input,
|
|
193 |
## Get DLC model and labels as strings
|
194 |
# TODO: make a dict as for megadetector
|
195 |
# pdb.set_trace()
|
196 |
-
|
197 |
-
|
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:
|
|
|
193 |
## Get DLC model and labels as strings
|
194 |
# TODO: make a dict as for megadetector
|
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:
|