multimodalart HF Staff commited on
Commit
52010af
·
verified ·
1 Parent(s): dd6a4cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.