Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
23 |
elif gender == 'Female':
|
24 |
-
return gr.Dropdown
|
25 |
else:
|
26 |
-
return gr.Dropdown
|
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):
|