AshanGimhana commited on
Commit
7ee0ce6
·
verified ·
1 Parent(s): 3d1f7d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -19,11 +19,11 @@ FACE_ANALYSER.prepare(ctx_id=0 if cv2.cuda.getCudaEnabledDeviceCount() > 0 else
19
  # Function to update template choices based on gender
20
  def update_templates(gender):
21
  if gender == 'Male':
22
- return gr.Dropdown.update(choices=['Boy Template 1.JPG', 'Boy Template 2.JPG', 'Boy Template 3.JPG'])
23
  elif gender == 'Female':
24
- return gr.Dropdown.update(choices=['Girl Template 1.JPG', 'Girl Template 2.JPG', 'Girl Template 3.JPG'])
25
  else:
26
- return gr.Dropdown.update(choices=[])
27
 
28
  # Main function for face swapping
29
  def face_swap_and_merge(src_image, gender, template_choice):
 
19
  # Function to update template choices based on gender
20
  def update_templates(gender):
21
  if gender == 'Male':
22
+ return gr.Dropdown(choices=['Boy Template 1.JPG', 'Boy Template 2.JPG', 'Boy Template 3.JPG'],interactive=True)
23
  elif gender == 'Female':
24
+ return gr.Dropdown(choices=['Girl Template 1.JPG', 'Girl Template 2.JPG', 'Girl Template 3.JPG'], interactive=True)
25
  else:
26
+ return gr.Dropdown(choices=[], interactive=True)
27
 
28
  # Main function for face swapping
29
  def face_swap_and_merge(src_image, gender, template_choice):