YinuoGuo27 commited on
Commit
c40d13d
·
verified ·
1 Parent(s): 2309cd3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -12
app.py CHANGED
@@ -32,6 +32,8 @@ def set_microphone():
32
  return "microphone"
33
  def set_tts():
34
  return "tts"
 
 
35
 
36
  @spaces.GPU
37
  def predict(prompt, upload_reference_audio, microphone_reference_audio, reference_audio_type):
@@ -111,19 +113,13 @@ def main():
111
  kd_submit = gr.Button("Generate", elem_id="kdtalker_generate", variant="primary")
112
 
113
  kd_submit.click(
114
- fn=kd_talker.generate_with_audio_img,
115
- inputs=[
116
- upload_driven_audio,
117
- tts_driven_audio,
118
- driven_audio_type,
119
- source_image,
120
- smoothed_pitch,
121
- smoothed_yaw,
122
- smoothed_roll,
123
- smoothed_t,
124
  ],
125
- outputs=[gen_video]
126
- )
127
  return interface
128
 
129
  if __name__ == "__main__":
 
32
  return "microphone"
33
  def set_tts():
34
  return "tts"
35
+ def create_kd_talker():
36
+ return Inferencer()
37
 
38
  @spaces.GPU
39
  def predict(prompt, upload_reference_audio, microphone_reference_audio, reference_audio_type):
 
113
  kd_submit = gr.Button("Generate", elem_id="kdtalker_generate", variant="primary")
114
 
115
  kd_submit.click(
116
+ fn=lambda *args: create_kd_talker().generate_with_audio_img(*args),
117
+ inputs=[
118
+ upload_driven_audio, tts_driven_audio, driven_audio_type, source_image,
119
+ smoothed_pitch, smoothed_yaw, smoothed_roll, smoothed_t
 
 
 
 
 
 
120
  ],
121
+ outputs=[gen_video]
122
+ )
123
  return interface
124
 
125
  if __name__ == "__main__":