mwmathis commited on
Commit
8ad13ed
Β·
1 Parent(s): cba3ae5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -12
app.py CHANGED
@@ -25,19 +25,10 @@ MD_models_dict = {'md_v5a': "MD_models/md_v5a.0.0.pt", #
25
  'md_v5b': "MD_models/md_v5b.0.0.pt"}
26
 
27
  # DLC models target dirs
28
- DLC_models_dict = {#'full_cat': "DLC_models/DLC_Cat/",
29
- #'full_dog': "DLC_models/DLC_Dog/",
30
- 'full_human': "DLC_models/DLC_human_dancing/",
31
- 'full_macaque': 'DLC_models/DLC_monkey/',
32
- 'primate_face': "DLC_models/DLC_FacialLandmarks/"}
33
 
34
-
35
- # FONTS = {'amiko': "fonts/Amiko-Regular.ttf",
36
- # 'nature': "fonts/LoveNature.otf",
37
- # 'painter':"fonts/PainterDecorator.otf",
38
- # 'animals': "fonts/UncialAnimals.ttf",
39
- # 'zen': "fonts/ZEN.TTF"}
40
- #####################################################
41
  def predict_pipeline(img_input,
42
  mega_model_input,
43
  dlc_model_input_str,
 
25
  'md_v5b': "MD_models/md_v5b.0.0.pt"}
26
 
27
  # DLC models target dirs
28
+ DLC_models_dict = {'supernanimal_quadruped': "DLC_models/DLC_Cat/",
29
+ 'supernanimal_topviewmouse': "DLC_models/DLC_Dog/"
30
+ }
 
 
31
 
 
 
 
 
 
 
 
32
  def predict_pipeline(img_input,
33
  mega_model_input,
34
  dlc_model_input_str,