hpoghos commited on
Commit
22aeb46
·
1 Parent(s): 90e4a07
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -86,7 +86,7 @@ def generate(prompt, num_frames, image, model_name_stage1, model_name_stage2, se
86
  video_path = opj(where_to_log, name+".mp4")
87
  return video_path
88
 
89
- @spaces.GPU(duration=360)
90
  def enhance(prompt, input_to_enhance, num_frames=None, image=None, model_name_stage1=None, model_name_stage2=None, seed=33, t=50, image_guidance=9.5, result_fol=result_fol):
91
  if input_to_enhance is None:
92
  input_to_enhance = generate(prompt, num_frames, image, model_name_stage1, model_name_stage2, seed, t, image_guidance)
 
86
  video_path = opj(where_to_log, name+".mp4")
87
  return video_path
88
 
89
+ @spaces.GPU(duration=300)
90
  def enhance(prompt, input_to_enhance, num_frames=None, image=None, model_name_stage1=None, model_name_stage2=None, seed=33, t=50, image_guidance=9.5, result_fol=result_fol):
91
  if input_to_enhance is None:
92
  input_to_enhance = generate(prompt, num_frames, image, model_name_stage1, model_name_stage2, seed, t, image_guidance)