Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
115 |
-
|
116 |
-
upload_driven_audio,
|
117 |
-
|
118 |
-
driven_audio_type,
|
119 |
-
source_image,
|
120 |
-
smoothed_pitch,
|
121 |
-
smoothed_yaw,
|
122 |
-
smoothed_roll,
|
123 |
-
smoothed_t,
|
124 |
],
|
125 |
-
|
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__":
|