Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 = {
|
| 29 |
-
|
| 30 |
-
|
| 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,
|