Spaces:
Running
on
Zero
Running
on
Zero
remove model loading on startup
Browse files
app.py
CHANGED
@@ -28,6 +28,9 @@ os.environ['NUMBA_THREADING_LAYER'] = 'workqueue'
|
|
28 |
download_scenedino_checkpoint("ssc-kitti-360-dino")
|
29 |
download_scenedino_checkpoint("ssc-kitti-360-dinov2")
|
30 |
|
|
|
|
|
|
|
31 |
|
32 |
def convert_voxels(arr, map_dict):
|
33 |
f = np.vectorize(map_dict.__getitem__)
|
@@ -61,6 +64,9 @@ def demo_run(image: str,
|
|
61 |
y_range: int,
|
62 |
z_range: int):
|
63 |
|
|
|
|
|
|
|
64 |
if backbone == "DINO (ViT-B)":
|
65 |
net, renderer, ray_sampler = net_v1, renderer_v1, ray_sampler_v1
|
66 |
elif backbone == "DINOv2 (ViT-B)":
|
@@ -191,10 +197,6 @@ def demo_run(image: str,
|
|
191 |
return output_2d, glb_path
|
192 |
|
193 |
|
194 |
-
net_v1, renderer_v1, ray_sampler_v1 = None, None, None
|
195 |
-
net_v2, renderer_v2, ray_sampler_v2 = None, None, None
|
196 |
-
load_scenedino()
|
197 |
-
|
198 |
demo = gr.Interface(
|
199 |
demo_run,
|
200 |
inputs=[
|
|
|
28 |
download_scenedino_checkpoint("ssc-kitti-360-dino")
|
29 |
download_scenedino_checkpoint("ssc-kitti-360-dinov2")
|
30 |
|
31 |
+
net_v1, renderer_v1, ray_sampler_v1 = None, None, None
|
32 |
+
net_v2, renderer_v2, ray_sampler_v2 = None, None, None
|
33 |
+
|
34 |
|
35 |
def convert_voxels(arr, map_dict):
|
36 |
f = np.vectorize(map_dict.__getitem__)
|
|
|
64 |
y_range: int,
|
65 |
z_range: int):
|
66 |
|
67 |
+
if net_v1 is None:
|
68 |
+
load_scenedino()
|
69 |
+
|
70 |
if backbone == "DINO (ViT-B)":
|
71 |
net, renderer, ray_sampler = net_v1, renderer_v1, ray_sampler_v1
|
72 |
elif backbone == "DINOv2 (ViT-B)":
|
|
|
197 |
return output_2d, glb_path
|
198 |
|
199 |
|
|
|
|
|
|
|
|
|
200 |
demo = gr.Interface(
|
201 |
demo_run,
|
202 |
inputs=[
|