sabrinabenas commited on
Commit
fb5441b
β€’
1 Parent(s): c4387de
Files changed (2) hide show
  1. app.py +2 -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, '/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:
 
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("/model/pretrained_model_urls.yaml") #FIXME
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]