a

app.py

@@ -247,6 +247,7 @@ def process_video(
     window_size,
     overlap
 ):
+    print('process_video called')
     try:
         pipe = get_model(model_size)
         if pipe is None:
@@ -645,8 +646,7 @@ with gr.Blocks(css=css, title="DKT", head=head_html) as demo:
             input_video, model_size, num_inference_steps, overlap
         ],
         outputs=[
-            output_video, vis_video, 
-            output_point_map0, output_point_map1, output_point_map2, output_point_map3
+            output_video, vis_video, output_point_map0, output_point_map1, output_point_map2, output_point_map3
         ]
     )
     
@@ -654,6 +654,8 @@ with gr.Blocks(css=css, title="DKT", head=head_html) as demo:
     
     example_files = glob.glob('examples/*')
     logger.info(f'there are {len(example_files)} demo files')
+    print(f'there are {len(example_files)} demo files')
+    
     if example_files:
         example_inputs = []
         for file_path in example_files:
@@ -676,9 +678,12 @@ if __name__ == '__main__':
     #* main code, model and moge model initialization
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     logger.info(f"device = {device}")
+    print(f"device = {device}")
+    
     load_model_1_3b(device=device)
     load_moge_model(device=device)
     torch.cuda.empty_cache()
+    print('model init done!')
 
     demo.queue().launch(share = False,server_name="0.0.0.0", server_port=7860)