imseldrith commited on
Commit
5d9f90a
1 Parent(s): c9a2adf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -21,7 +21,7 @@ def run(*args):
21
  selected_frame_processors = ' '.join(frame_processor)
22
 
23
  face_analyser_direction = rest_args[1]
24
- face_analyser_age = rest_args[2]
25
  face_analyser_gender = rest_args[3]
26
 
27
  cmd = (
@@ -29,8 +29,8 @@ def run(*args):
29
  f"--frame-processors {selected_frame_processors} "
30
  f"--face-analyser-direction {face_analyser_direction} "
31
  )
32
- if face_analyser_age != 'none':
33
- cmd += f" --face-analyser-age {face_analyser_age}"
34
  if face_analyser_gender != 'none':
35
  cmd += f" --face-analyser-gender {face_analyser_gender}"
36
 
@@ -120,9 +120,9 @@ with gr.Blocks(theme=get_theme(), title="DeepFakeAI 1.0.0") as ui:
120
  value = 'left-right'
121
  )
122
  face_analyser_age_dropdown = gr.Dropdown(
123
- label = 'FACE ANALYSER AGE',
124
- choices = ['none'] + ['child', 'teen', 'adult', 'senior'],
125
- value = 'none'
126
  )
127
  face_analyser_gender_dropdown = gr.Dropdown(
128
  label = 'FACE ANALYSER GENDER',
 
21
  selected_frame_processors = ' '.join(frame_processor)
22
 
23
  face_analyser_direction = rest_args[1]
24
+ face_recognition = rest_args[2]
25
  face_analyser_gender = rest_args[3]
26
 
27
  cmd = (
 
29
  f"--frame-processors {selected_frame_processors} "
30
  f"--face-analyser-direction {face_analyser_direction} "
31
  )
32
+ if face_recognition != 'none':
33
+ cmd += f" --face-recognition {face_recognition}"
34
  if face_analyser_gender != 'none':
35
  cmd += f" --face-analyser-gender {face_analyser_gender}"
36
 
 
120
  value = 'left-right'
121
  )
122
  face_analyser_age_dropdown = gr.Dropdown(
123
+ label = 'FACE RECOGNITION',
124
+ choices = ['none'] + ['reference', 'many'],
125
+ value = 'reference'
126
  )
127
  face_analyser_gender_dropdown = gr.Dropdown(
128
  label = 'FACE ANALYSER GENDER',