download

app.py

@@ -31,13 +31,19 @@ FONTS = {'amiko': "font/Amiko-Regular.ttf",
 Megadet_Models = {'md_v5a': "megadet_model/md_v5a.0.0.pt",
                   'md_v5b': "megadet_model/md_v5b.0.0.pt"}
 
+DLC_folders = {'full_cat': "model/DLC_Cat/",
+              'full_dog': "model/DLC_Dog/",
+              'primate_face': "model/DLC_FacialLandmarks/",
+              'full_human': "model/DLC_human_dancing/",
+              'full_macaque': 'model/DLC_monkey/'}
+
 DLC_models = {'full_cat': "model/DLC_Cat_resnet_50_iteration-0_shuffle-0",
               'full_dog': "model/DLC_Dog_resnet_50_iteration-0_shuffle-0",
               'primate_face': "model/DLC_FacialLandmarks_resnet_50_iteration-1_shuffle-1",
               'full_human': "model/DLC_human_dancing_resnet_101_iteration-0_shuffle-1",
-              'full_macaque': 'model/DLC_monkey_resnet_50_iteration-0_shuffle-1'}      
+              'full_macaque': 'model/DLC_monkey_resnet_50_iteration-0_shuffle-1'}
 
-DLC_models_list = ['full_cat', 'full_dog','full_human', 'full_macaque', 'primate_face']  
+DLC_models_list = ['full_cat', 'full_dog','primate_face', 'full_human', 'full_macaque']
            
 #########################################
 #  Draw keypoints on image
@@ -307,7 +313,7 @@ gr_slider_conf_keypoints = gr.inputs.Slider(0,1,.05,0.25,
                                             label='Set confidence threshold for keypoints')
 
 # Data viz                                            
-gr_keypt_color = gr.ColorPicker(value ="#ff0000", label="choose color for keypoint label") 
+gr_keypt_color = gr.ColorPicker(label="choose color for keypoint label") 
 
 gr_labels_font_style = gr.inputs.Dropdown(choices=['amiko', 'nature', 'painter', 'animals', 'zen'],
                                         default='amiko',