add spend time calc

app.py

@@ -148,11 +148,24 @@ def process_video(
     
 
 
-    prediction_result = DKT_PIPELINE(video_file, prompt=PROMPT, \
-        negative_prompt=NEGATIVE_PROMPT,\
-            height=height,width=width,num_inference_steps=num_inference_steps,\
-                overlap=overlap, return_rgb=True)
-    
+    import time
+    start_time = time.time()
+
+    prediction_result = DKT_PIPELINE(
+        video_file,
+        prompt=PROMPT,
+        negative_prompt=NEGATIVE_PROMPT,
+        height=height,
+        width=width,
+        num_inference_steps=num_inference_steps,
+        overlap=overlap,
+        return_rgb=True
+    )
+
+    end_time = time.time()
+    spend_time = end_time - start_time
+    logger.info(f"DKT_PIPELINE spend time: {spend_time:.2f} seconds for depth prediction")
+
     
     
 
@@ -162,7 +175,11 @@ def process_video(
     indices = np.round(indices).astype(np.int32)
     
   
+    pc_start_time = time.time()
     pcds = DKT_PIPELINE.prediction2pc_v2(prediction_result['depth_map'], prediction_result['rgb_frames'], indices, return_pcd=True)
+    pc_end_time = time.time()
+    pc_spend_time = pc_end_time - pc_start_time
+    logger.info(f"prediction2pc_v2 spend time: {pc_spend_time:.2f} seconds for point cloud extraction")
     glb_files = []
 
     for idx, pcd in enumerate(pcds):
@@ -418,9 +435,13 @@ with gr.Blocks(css=css, title="DKT", head=head_html) as demo:
             return None, None, None, None, None, None, "Please upload a video file"
         
         try:
+            import time
+            start_time = time.time()
             output_path, glb_files = process_video(
                 video_file, model_size, num_inference_steps, overlap
             )
+            spend_time = time.time() - start_time
+            logger.info(f"Total spend time in on_submit: {spend_time:.2f} seconds")
 
             
             if output_path is None: