diff --git a/demo/ui.py b/demo/ui.py index 872ea97c..cd7209c4 100644 --- a/demo/ui.py +++ b/demo/ui.py @@ -16,12 +16,15 @@ def create_ui( ): DEFAULT_LANG = "zh" DEFAULT_HUMAN_MATTING_MODEL = "modnet_photographic_portrait_matting" - DEFAULT_FACE_DETECT_MODEL = "mtcnn" + DEFAULT_FACE_DETECT_MODEL = "retinaface-resnet50" if DEFAULT_HUMAN_MATTING_MODEL in human_matting_models: human_matting_models.remove(DEFAULT_HUMAN_MATTING_MODEL) human_matting_models.insert(0, DEFAULT_HUMAN_MATTING_MODEL) + if DEFAULT_FACE_DETECT_MODEL not in face_detect_models: + DEFAULT_FACE_DETECT_MODEL = "mtcnn" + demo = gr.Blocks(title="HivisionIDPhotos") with demo: