open3d fix

app.py

@@ -7,21 +7,22 @@ import sys
 import yaml
 
 import numpy as np
-#import open3d as o3d
+import open3d_fix
+import open3d as o3d
 
 import gradio as gr
 import spaces
 import torch
 
-#sys.path.append("./sscbench")
-#from sscbench.gen_voxelgrid_npy import save_as_voxel_ply, classes_to_colors
+sys.path.append("./sscbench")
+from sscbench.gen_voxelgrid_npy import save_as_voxel_ply, classes_to_colors
 from download_checkpoint_hf import download_scenedino_checkpoint
 
-#from demo_utils.utils import (load_modules, 
-#                              load_sample_from_path, 
-#                              get_fov_mask,
-#                              inference_3d, 
-#                              inference_rendered_2d)
+from demo_utils.utils import (load_modules, 
+                              load_sample_from_path, 
+                              get_fov_mask,
+                              inference_3d, 
+                              inference_rendered_2d)
 
 
 # Load checkpoints from Hugging Face
@@ -64,8 +65,6 @@ def demo_run(image: str,
              y_range: int, 
              z_range: int):
 
-    return image
-
     if net_v1 is None:
         load_scenedino()
 

open3d_fix.py

@@ -0,0 +1,9 @@
+# Fix as suggested in https://huggingface.co/spaces/TencentARC/FreeSplatter/discussions/3
+
+import fileinput
+import site
+from pathlib import Path
+
+with fileinput.FileInput(f'{site.getsitepackages()[0]}/open3d/__init__.py', inplace=True) as file:
+    for line in file:
+        print(line.replace('_pybind_cuda.open3d_core_cuda_device_count()', '1'), end='')