fix model loading

app.py

@@ -29,8 +29,17 @@ from demo_utils.utils import (load_modules,
 download_scenedino_checkpoint("ssc-kitti-360-dino")
 download_scenedino_checkpoint("ssc-kitti-360-dinov2")
 
-net_v1, renderer_v1, ray_sampler_v1 = None, None, None
-net_v2, renderer_v2, ray_sampler_v2 = None, None, None
+
+# Load model, ray sampler, datasets
+ckpt_path = "out/scenedino-pretrained/seg-best-dino/"
+ckpt_name = "checkpoint.pt"
+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)
+renderer_v2.eval()
 
 
 def convert_voxels(arr, map_dict):
@@ -41,20 +50,6 @@ with open("sscbench/label_maps.yaml", "r") as f:
     label_maps = yaml.safe_load(f)
 
 
-@spaces.GPU
-def load_scenedino():
-    # Load model, ray sampler, datasets
-    ckpt_path = "out/scenedino-pretrained/seg-best-dino/"
-    ckpt_name = "checkpoint.pt"
-    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)
-    renderer_v2.eval()
-
-
 @spaces.GPU
 def demo_run(image: str, 
              backbone: str,
@@ -65,9 +60,6 @@ def demo_run(image: str,
              y_range: int, 
              z_range: int):
 
-    if net_v1 is None:
-        load_scenedino()
-
     if backbone == "DINO (ViT-B)":
         net, renderer, ray_sampler = net_v1, renderer_v1, ray_sampler_v1
     elif backbone == "DINOv2 (ViT-B)":