remove dataset loading in demo

app.py

@@ -29,12 +29,12 @@ download_scenedino_checkpoint("ssc-kitti-360-dinov2")
 # Load model, ray sampler, datasets
 ckpt_path = "out/scenedino-pretrained/seg-best-dino/"
 ckpt_name = "checkpoint.pt"
-net_v1, renderer_v1, ray_sampler_v1, test_dataset = load_modules(ckpt_path, ckpt_name)
+net_v1, renderer_v1, ray_sampler_v1 = load_modules(ckpt_path, ckpt_name)
 renderer_v1.eval()
 
 ckpt_path = "out/scenedino-pretrained/seg-best-dinov2/"
 ckpt_name = "checkpoint.pt"
-net_v2, renderer_v2, ray_sampler_v2, _ = load_modules(ckpt_path, ckpt_name)
+net_v2, renderer_v2, ray_sampler_v2 = load_modules(ckpt_path, ckpt_name)
 renderer_v2.eval()
 
 

demo_utils/utils.py

@@ -55,9 +55,9 @@ def load_modules(
     model.load_state_dict(cp, strict=False)
     model = model.to(device)
 
-    test_dataset = make_datasets(config["dataset"])[1]
+    # test_dataset = make_datasets(config["dataset"])[1]
 
-    return net, renderer, ray_sampler, test_dataset
+    return net, renderer, ray_sampler  # , test_dataset
 
 
 def load_sample_from_path(