We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 28e861e commit 6040d92Copy full SHA for 6040d92
demo/ui.py
@@ -16,12 +16,15 @@ def create_ui(
16
):
17
DEFAULT_LANG = "zh"
18
DEFAULT_HUMAN_MATTING_MODEL = "modnet_photographic_portrait_matting"
19
- DEFAULT_FACE_DETECT_MODEL = "mtcnn"
+ DEFAULT_FACE_DETECT_MODEL = "retinaface-resnet50"
20
21
if DEFAULT_HUMAN_MATTING_MODEL in human_matting_models:
22
human_matting_models.remove(DEFAULT_HUMAN_MATTING_MODEL)
23
human_matting_models.insert(0, DEFAULT_HUMAN_MATTING_MODEL)
24
25
+ if DEFAULT_FACE_DETECT_MODEL not in face_detect_models:
26
+ DEFAULT_FACE_DETECT_MODEL = "mtcnn"
27
+
28
demo = gr.Blocks(title="HivisionIDPhotos")
29
30
with demo:
0 commit comments