Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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=
|
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
|