sabrinabenas commited on
Commit
668cb3e
β€’
1 Parent(s): 38d9537

test download

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- #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:
 
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: