Update app.py
Browse files
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 =
|
89 |
-
|
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],
|