Update app.py

app.py

@@ -10,17 +10,31 @@ import transformers
 from PIL import Image
 import requests
 
+from viz_utils import save_results_as_json, draw_keypoints_on_image, draw_bbox_w_text, save_results_only_dlc
+from detection_utils import predict_md, crop_animal_detections
+from ui_utils import gradio_inputs_for_MD_DLC, gradio_outputs_for_MD_DLC, gradio_description_and_examples
+
 from deeplabcut.utils import auxiliaryfunctions
 from dlclibrary.dlcmodelzoo.modelzoo_download import (
     download_huggingface_model,
     MODELOPTIONS,
 )
 
+# megadetector and dlc model look up
+MD_models_dict = {'md_v5a': "MD_models/md_v5a.0.0.pt", # 
+                  'md_v5b': "MD_models/md_v5b.0.0.pt"}
+
+# DLC models target  dirs
+DLC_models_dict = {'superanimal_topviewmouse': "DLC_models/sa-tvm",
+                   'superanimal_quadreped': "DLC_models/sa-q",
+                    'full_human': "DLC_models/DLC_human_dancing/"}
+
+# download the SuperAnimal models:
 model = 'superanimal_topviewmouse'
 train_dir = 'DLC_models/sa-tvm'
 download_huggingface_model(model, train_dir)
 
-#demo data cooco cat:
+# grab demo data cooco cat:
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)