upd

app.py

@@ -39,7 +39,7 @@ def check_video_input(image_block: Image.Image):
         raise gr.Error("Please generate a video first")
 
 
-@spaces.GPU(duration=120)
+@spaces.GPU()
 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')
@@ -56,14 +56,14 @@ def optimize_stage_1(image_block: Image.Image, preprocess_chk: bool, seed_slider
 
     # stage 1
     subprocess.run(f'export MKL_THREADING_LAYER=GNU;export MKL_SERVICE_FORCE_INTEL=1;python scripts/gen_vid.py --path tmp_data/{img_hash}_rgba.png --seed {seed_slider} --bg white', shell=True)
-
+    subprocess.run(f'python lgm/infer.py big --resume {ckpt_path} --test_path tmp_data/{img_hash}_rgba.png', shell=True)
     # return [os.path.join('logs', 'tmp_rgba_model.ply')]
     return os.path.join('tmp_data', f'{img_hash}_rgba_generated.mp4')
 
 @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)
+
     # stage 2
     subprocess.run(f'python main_4d.py --config {os.path.join("configs", "4d_demo.yaml")} input={os.path.join("tmp_data", f"{img_hash}_rgba.png")}', shell=True)
     # os.rename(os.path.join('logs', f'{img_hash}_rgba_frames'), os.path.join('logs', f'{img_hash}_{seed_slider:03d}_rgba_frames'))