Update app.py

app.py

@@ -66,7 +66,19 @@ def predict_pipeline(img_input,
                                             md_results,
                                             bbox_likelihood_th)
 
-    ############################################################
+        ############################################################
 
+############################################################
+    ## Get DLC model and label map  
+    
+    # If model is found: do not download (previous execution is likely within same day)
+    # TODO: can we ask the user whether to reload dlc model if a directory is found?
+    if os.path.isdir(DLC_models_dict[dlc_model_input_str]) and \
+        len(os.listdir(DLC_models_dict[dlc_model_input_str])) > 0:
+        path_to_DLCmodel = DLC_models_dict[dlc_model_input_str]
+    else:
+        path_to_DLCmodel = download_huggingface_model(dlc_model_input_str, 
+                                         DLC_models_dict[dlc_model_input_str])
 
+