jiaweir commited on
Commit
56c83bd
β€’
1 Parent(s): 1e3bb59
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def check_video_input(image_block: Image.Image):
39
  raise gr.Error("Please generate a video first")
40
 
41
 
42
- @spaces.GPU()
43
  def optimize_stage_1(image_block: Image.Image, preprocess_chk: bool, seed_slider: int):
44
  if not os.path.exists('tmp_data'):
45
  os.makedirs('tmp_data')
 
39
  raise gr.Error("Please generate a video first")
40
 
41
 
42
+ @spaces.GPU(duration=120)
43
  def optimize_stage_1(image_block: Image.Image, preprocess_chk: bool, seed_slider: int):
44
  if not os.path.exists('tmp_data'):
45
  os.makedirs('tmp_data')