Update app.py

app.py

@@ -7,12 +7,15 @@ import yaml
 import numpy as np
 from matplotlib import cm
 import gradio as gr
+import deeplabcut
+import dlclib
 
 from PIL import Image, ImageColor, ImageFont, ImageDraw 
 # check git lfs pull!!
 from DLC_models.download_utils import DownloadModel
 from dlclive import DLCLive, Processor
 
+
 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, predict_dlc
 from ui_utils import gradio_inputs_for_MD_DLC, gradio_outputs_for_MD_DLC, gradio_description_and_examples
@@ -25,8 +28,8 @@ 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 = {'supernanimal_quadruped': "DLC_models/DLC_Cat/",
-                    'supernanimal_topviewmouse': "DLC_models/DLC_Dog/"
+DLC_models_dict = {'supernanimal_quadruped': "DLC_models/supernanimal_quadruped/",
+                    'supernanimal_topviewmouse': "DLC_models/supernanimal_topviewmouse/"
                     }
                  
 def predict_pipeline(img_input,