narugo commited on
Commit
aa7e786
·
verified ·
1 Parent(s): fb39199

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -29,8 +29,11 @@ if __name__ == '__main__':
29
  YOLOModel('deepghs/yolo-face').make_ui(
30
  default_model_name='yolov10s-face',
31
  )
32
- # with gr.Tab('Head Detection'):
33
- # HeadDetection().make_ui()
 
 
 
34
  # with gr.Tab('Person Detection'):
35
  # PersonDetection().make_ui()
36
  with gr.Tab('Generic'):
 
29
  YOLOModel('deepghs/yolo-face').make_ui(
30
  default_model_name='yolov10s-face',
31
  )
32
+ with gr.Tab('Head Detection'):
33
+ YOLOModel('deepghs/real_head_detection').make_ui(
34
+ default_model_name='head_detect_v0_s_yv11',
35
+ default_conf_threshold=0.18,
36
+ )
37
  # with gr.Tab('Person Detection'):
38
  # PersonDetection().make_ui()
39
  with gr.Tab('Generic'):