YinuoGuo27 commited on
Commit
3ac881c
·
verified ·
1 Parent(s): 5c7919e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -50,6 +50,9 @@ def predict(prompt, upload_reference_audio, microphone_reference_audio, referenc
50
  )
51
  return gr.Audio(value=output_file_path, type='filepath')
52
 
 
 
 
53
  @spaces.GPU
54
  def main():
55
  if torch.cuda.is_available():
@@ -108,7 +111,7 @@ def main():
108
  smoothed_t = gr.Slider(minimum=0, maximum=1, step=0.1, label="T", value=0.8)
109
  kd_submit = gr.Button("Generate", elem_id="kdtalker_generate", variant="primary")
110
  kd_submit.click(
111
- fn=Inferencer().generate_with_audio_img,
112
  inputs=[
113
  upload_driven_audio, tts_driven_audio, driven_audio_type, source_image,
114
  smoothed_pitch, smoothed_yaw, smoothed_roll, smoothed_t
 
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():
58
  if torch.cuda.is_available():
 
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