longer duration

app.py

@@ -310,7 +310,7 @@ def get_duration(rgb_image, seg_image, seed, randomize_seed,
                  cfg_interval_end, t_rescale, work_space):
     instance_labels = np.unique(np.array(seg_image).reshape(-1, 3), axis=0)
     step_duration = 15.0
-    return instance_labels.shape[0] * step_duration + 180
+    return instance_labels.shape[0] * step_duration + 240
 
 @spaces.GPU(duration=get_duration)
 def run_generation(
@@ -324,7 +324,8 @@ def run_generation(
     cfg_interval_end: float = 1.0,
     t_rescale: float = 3.0,
     work_space: dict = None,
-):
+):  
+    first_render = True
     if work_space is None:
         raise gr.Error("Please run step 1 and step 2 first.")
     required_keys = ["dir", "depth_mask", "depth", "K", "c2w", "trans", "scale"]
@@ -388,10 +389,13 @@ def run_generation(
     run_id = str(uuid.uuid4())
 
     DEVICE = 'cuda'
+    gr.Info('Loading ckpts')
+    down_t = time.time()
     pipeline = ThreeDFixerPipeline.from_pretrained(
         local_dir, compile=False
     )
     pipeline.to(device=DEVICE)
+    gr.Info(f'Loading ckpts duration: {time.time()-down_t:.2}s')
 
     if not isinstance(rgb_image, Image.Image) and "image" in rgb_image:
         rgb_image = rgb_image["image"]
@@ -644,10 +648,16 @@ def run_generation(
                     os.path.join(work_space['dir'], f"{run_id}_instance_gs_fine_{instance_name}.png")
                 )
 
+
+                if first_render:
+                    render_t = time.time()
                 video = render_utils.render_video(
                     outputs["gaussian"][0],
                     bg_color=(1.0, 1.0, 1.0)
                 )["color"]
+                if first_render:
+                    gr.Info(f'Compile gsplat duration: {time.time()-render_t:.2}s')
+                    first_render = False
                 imageio.mimsave(mp4_path, video, fps=30)
                 imageio.imwrite(poster_path, video[0])