mwmathis commited on
Commit
b11b05c
Β·
1 Parent(s): 3e17a60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -1
app.py CHANGED
@@ -66,7 +66,19 @@ def predict_pipeline(img_input,
66
  md_results,
67
  bbox_likelihood_th)
68
 
69
- ############################################################
70
 
 
 
 
 
 
 
 
 
 
 
 
71
 
 
72
 
 
66
  md_results,
67
  bbox_likelihood_th)
68
 
69
+ ############################################################
70
 
71
+ ############################################################
72
+ ## Get DLC model and label map
73
+
74
+ # If model is found: do not download (previous execution is likely within same day)
75
+ # TODO: can we ask the user whether to reload dlc model if a directory is found?
76
+ if os.path.isdir(DLC_models_dict[dlc_model_input_str]) and \
77
+ len(os.listdir(DLC_models_dict[dlc_model_input_str])) > 0:
78
+ path_to_DLCmodel = DLC_models_dict[dlc_model_input_str]
79
+ else:
80
+ path_to_DLCmodel = download_huggingface_model(dlc_model_input_str,
81
+ DLC_models_dict[dlc_model_input_str])
82
 
83
+
84