jev-aleks commited on
Commit
26438d4
·
1 Parent(s): 7abe500

remove model loading on startup

Browse files
Files changed (1) hide show
  1. app.py +6 -4
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=[