fix output of inference

app.py

@@ -129,9 +129,10 @@ def run_model(target_dir, model) -> dict:
 
     # Generate world points from depth map
     print("Computing world points from depth map...")
-    depth_map = predictions["depth"]  # (S, H, W, 1)
-    world_points = unproject_depth_map_to_point_map(depth_map, predictions["extrinsic"], predictions["intrinsic"])
-    predictions["world_points_from_depth"] = world_points
+    #depth_map = predictions["depth"]  # (S, H, W, 1)
+    #world_points = unproject_depth_map_to_point_map(depth_map, predictions["extrinsic"], predictions["intrinsic"])
+    #predictions["world_points_from_depth"] = world_points
+    predictions["world_points_from_depth"] = predictions["world_points"]
 
     # Clean up
     torch.cuda.empty_cache()