Update app.py
Browse files
app.py
CHANGED
@@ -120,7 +120,7 @@ except Exception as e:
|
|
120 |
emo_name_to_id = {v: k for k, v in emo_map.items()}
|
121 |
|
122 |
# --- Core Generation Function ---
|
123 |
-
@spaces.GPU
|
124 |
def generate_motion(source_image_path, driving_audio_path, emotion_name, cfg_scale, progress=gr.Progress(track_tqdm=True)):
|
125 |
"""
|
126 |
The main function that takes Gradio inputs and generates the talking head video.
|
|
|
120 |
emo_name_to_id = {v: k for k, v in emo_map.items()}
|
121 |
|
122 |
# --- Core Generation Function ---
|
123 |
+
@spaces.GPU(duration=120)
|
124 |
def generate_motion(source_image_path, driving_audio_path, emotion_name, cfg_scale, progress=gr.Progress(track_tqdm=True)):
|
125 |
"""
|
126 |
The main function that takes Gradio inputs and generates the talking head video.
|