fix output of inference

app.py

@@ -120,8 +120,8 @@ def run_model(target_dir, model) -> dict:
     # 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"].unsqueeze(0) if predictions["pose_enc"].ndim == 2 else predictions["pose_enc"], images.shape[-2:])
-    predictions["extrinsic"] = extrinsic
-    predictions["intrinsic"] = intrinsic
+    predictions["extrinsic"] = extrinsic.squeeze(0)  # (S, 3, 4)
+    predictions["intrinsic"] = intrinsic.squeeze(0) if intrinsic is not None else None  # (S, 3, 3) or None
     print("Extrinsic shape:", predictions["extrinsic"].shape)
     print("Intrinsic shape:", predictions["intrinsic"].shape)