jeylau commited on
Commit
ff4cf6b
Β·
1 Parent(s): a5e6f78

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -85,8 +85,8 @@ def predict_pipeline(img_input,
85
  len(os.listdir(DLC_models_dict[dlc_model_input_str])) > 0:
86
  path_to_DLCmodel = DLC_models_dict[dlc_model_input_str]
87
  else:
88
- path_to_DLCmodel = download_huggingface_model(dlc_model_input_str,
89
- DLC_models_dict[dlc_model_input_str])
90
 
91
  # extract map label ids to strings
92
  pose_cfg_path = os.path.join(DLC_models_dict[dlc_model_input_str],
 
85
  len(os.listdir(DLC_models_dict[dlc_model_input_str])) > 0:
86
  path_to_DLCmodel = DLC_models_dict[dlc_model_input_str]
87
  else:
88
+ path_to_DLCmodel = DLC_models_dict[dlc_model_input_str]
89
+ download_huggingface_model(dlc_model_input_str, path_to_DLCmodel)
90
 
91
  # extract map label ids to strings
92
  pose_cfg_path = os.path.join(DLC_models_dict[dlc_model_input_str],