abreza commited on
Commit
3615bea
1 Parent(s): 3dca45e

add duration

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,7 +32,7 @@ def mp3_to_wav(mp3_filename, wav_filename, frame_rate):
32
  frame_rate).export(wav_filename, format="wav")
33
 
34
 
35
- @spaces.GPU()
36
  def generate_video(source_image, driven_audio, preprocess='crop', still_mode=False, use_enhancer=False,
37
  batch_size=1, size=256, pose_style=0, facerender='facevid2vid', exp_scale=1.0,
38
  use_ref_video=False, ref_video=None, ref_info=None, use_idle_mode=False,
@@ -195,7 +195,7 @@ with gr.Blocks(analytics_enabled=False) as demo:
195
  batch_size = gr.Slider(
196
  label="batch size in generation", step=1, maximum=10, value=1)
197
  enhancer = gr.Checkbox(
198
- label="GFPGAN as Face enhancer")
199
 
200
  submit = gr.Button(
201
  'Generate', elem_id="sadtalker_generate", variant='primary')
 
32
  frame_rate).export(wav_filename, format="wav")
33
 
34
 
35
+ @spaces.GPU(duration=120)
36
  def generate_video(source_image, driven_audio, preprocess='crop', still_mode=False, use_enhancer=False,
37
  batch_size=1, size=256, pose_style=0, facerender='facevid2vid', exp_scale=1.0,
38
  use_ref_video=False, ref_video=None, ref_info=None, use_idle_mode=False,
 
195
  batch_size = gr.Slider(
196
  label="batch size in generation", step=1, maximum=10, value=1)
197
  enhancer = gr.Checkbox(
198
+ label="GFPGAN as Face enhancer", value=True)
199
 
200
  submit = gr.Button(
201
  'Generate', elem_id="sadtalker_generate", variant='primary')