YinuoGuo27 commited on
Commit
99167d3
·
verified ·
1 Parent(s): 513c6ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -50,8 +50,7 @@ def predict(prompt, upload_reference_audio, microphone_reference_audio, referenc
50
  )
51
  return gr.Audio(value=output_file_path, type='filepath')
52
 
53
- def create_inferencer():
54
- return Inferencer()
55
 
56
  @spaces.GPU
57
  def main():
@@ -110,8 +109,9 @@ def main():
110
  smoothed_roll = gr.Slider(minimum=0, maximum=1, step=0.1, label="Roll", value=0.8)
111
  smoothed_t = gr.Slider(minimum=0, maximum=1, step=0.1, label="T", value=0.8)
112
  kd_submit = gr.Button("Generate", elem_id="kdtalker_generate", variant="primary")
 
113
  kd_submit.click(
114
- fn=create_inferencer().generate_with_audio_img,
115
  inputs=[
116
  upload_driven_audio, tts_driven_audio, driven_audio_type, source_image,
117
  smoothed_pitch, smoothed_yaw, smoothed_roll, smoothed_t
 
50
  )
51
  return gr.Audio(value=output_file_path, type='filepath')
52
 
53
+
 
54
 
55
  @spaces.GPU
56
  def main():
 
109
  smoothed_roll = gr.Slider(minimum=0, maximum=1, step=0.1, label="Roll", value=0.8)
110
  smoothed_t = gr.Slider(minimum=0, maximum=1, step=0.1, label="T", value=0.8)
111
  kd_submit = gr.Button("Generate", elem_id="kdtalker_generate", variant="primary")
112
+ kdtalker = Inferencer()
113
  kd_submit.click(
114
+ fn=kdtalker.generate_with_audio_img,
115
  inputs=[
116
  upload_driven_audio, tts_driven_audio, driven_audio_type, source_image,
117
  smoothed_pitch, smoothed_yaw, smoothed_roll, smoothed_t