update

app.py

@@ -1,4 +1,6 @@
 import os
+import torch
+import sys
 import subprocess
 
 def install_cuda_toolkit():
@@ -20,21 +22,20 @@ def install_cuda_toolkit():
 
 install_cuda_toolkit() # to compile the dependencies
 
-os.system('pip install ./submodules/simple-knn')
-os.system('pip install ./submodules/diff-surfel-rasterization')
-# import simple_knn
-# import diff_surfel_rasterization
-
-import torch
 pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
 version_str="".join([
     f"py3{sys.version_info.minor}_cu",
     torch.version.cuda.replace(".",""),
     f"_pyt{pyt_version_str}"
 ])
+# install pytorch3d with the right version
 os.system('pip install iopath')
 os.system(f'pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html')
 
+# install 2dgs dependencies
+os.system('pip install ./submodules/simple-knn')
+os.system('pip install ./submodules/diff-surfel-rasterization')
+
 import argparse
 import spaces
 import json