upd

app.py

@@ -38,7 +38,8 @@ def check_video_input(image_block: Image.Image):
     if not os.path.exists(os.path.join('tmp_data', f'{img_hash}_rgba_generated.mp4')):
         raise gr.Error("Please generate a video first")
 
-@spaces.GPU
+
+@spaces.GPU(duration=120)
 def optimize_stage_1(image_block: Image.Image, preprocess_chk: bool, seed_slider: int):
     if not os.path.exists('tmp_data'):
         os.makedirs('tmp_data')
@@ -59,7 +60,7 @@ def optimize_stage_1(image_block: Image.Image, preprocess_chk: bool, seed_slider
     # return [os.path.join('logs', 'tmp_rgba_model.ply')]
     return os.path.join('tmp_data', f'{img_hash}_rgba_generated.mp4')
 
-@spaces.GPU
+@spaces.GPU(duration=180)
 def optimize_stage_2(image_block: Image.Image, seed_slider: int):
     img_hash = hashlib.sha256(image_block.tobytes()).hexdigest()
     subprocess.run(f'python lgm/infer.py big --resume {ckpt_path} --test_path tmp_data/{img_hash}_rgba.png', shell=True)