fffiloni commited on
Commit
a8ba4df
·
verified ·
1 Parent(s): 9c8c926

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import gradio as gr
 
2
  import os
3
  import sys
4
  import shutil
@@ -79,6 +80,7 @@ from accelerate.utils import set_seed
79
  from latentsync.whisper.audio2feature import Audio2Feature
80
 
81
 
 
82
  def main(video_path, audio_path, progress=gr.Progress(track_tqdm=True)):
83
  inference_ckpt_path = "checkpoints/latentsync_unet.pt"
84
  unet_config_path = "configs/unet/second_stage.yaml"
@@ -218,4 +220,4 @@ with gr.Blocks(css=css) as demo:
218
  outputs = [video_result]
219
  )
220
 
221
- demo.queue().launch(show_api=False, show_error=True)
 
1
  import gradio as gr
2
+ import spaces
3
  import os
4
  import sys
5
  import shutil
 
80
  from latentsync.whisper.audio2feature import Audio2Feature
81
 
82
 
83
+ @spaces.GPU(duration=180)
84
  def main(video_path, audio_path, progress=gr.Progress(track_tqdm=True)):
85
  inference_ckpt_path = "checkpoints/latentsync_unet.pt"
86
  unet_config_path = "configs/unet/second_stage.yaml"
 
220
  outputs = [video_result]
221
  )
222
 
223
+ demo.queue().launch(show_api=True, show_error=True, mcp_server=True)