Update app.py

app.py

@@ -35,7 +35,9 @@ def calculate(image_in, audio_in):
     return "/content/train/image_audio.mp4"
 
 def one_shot(image_in,audio_in):
-    return calculate(image_in,audio_in)
+    if gender == 'Female' or gender == 'female':
+        return calculate(image_in,audio_in
+                        
 def run():
   with block:
     
@@ -44,12 +46,13 @@ def run():
         with gr.Row().style(equal_height=True):
           image_in = gr.Image(show_label=False, type="filepath")
           audio_in = gr.Audio(show_label=False, type='filepath')
+          gender = gr.Radio(["Female","Male"],value="Female",label="Gender")
           video_out = gr.Video(show_label=False)
         with gr.Row().style(equal_height=True):
           btn = gr.Button("Generate")          
 
     
-    btn.click(one_shot, inputs=[image_in, audio_in], outputs=[video_out])
+    btn.click(one_shot, inputs=[image_in, audio_in,gender], outputs=[video_out])
     block.queue()
     block.launch(server_name="0.0.0.0", server_port=7860)