fix output of inference

app.py

@@ -73,6 +73,7 @@ def run_model(target_dir, model) -> dict:
 
     predictions = {}    
     predictions["images"] = images  # (S, 3, H, W)
+    print(f"Images shape: {images.shape}")
 
     frames = []
     for i in range(images.shape[0]):
@@ -114,7 +115,8 @@ def run_model(target_dir, model) -> dict:
     print("Depth map shape:", predictions["depth"].shape)
     print("Depth confidence shape:", predictions["depth_conf"].shape)
     print("Pose encoding shape:", predictions["pose_enc"].shape)
-
+    print(f"Images shape: {images.shape}")
+    
     # Convert pose encoding to extrinsic and intrinsic matrices
     print("Converting pose encoding to extrinsic and intrinsic matrices...")
     extrinsic, intrinsic = pose_encoding_to_extri_intri(predictions["pose_enc"], images.shape[-2:])