yslan commited on
Commit
266275c
1 Parent(s): 8d996ca
Files changed (2) hide show
  1. app.py +2 -3
  2. requirements.txt +2 -1
app.py CHANGED
@@ -21,10 +21,9 @@ def install_cuda_toolkit():
21
  install_cuda_toolkit() # to compile the dependencies
22
 
23
  os.system('pip install ./submodules/simple-knn')
24
- import simple_knn
25
-
26
  os.system('pip install ./submodules/diff-surfel-rasterization')
27
- import diff_surfel_rasterization
 
28
 
29
  import argparse
30
  import spaces
 
21
  install_cuda_toolkit() # to compile the dependencies
22
 
23
  os.system('pip install ./submodules/simple-knn')
 
 
24
  os.system('pip install ./submodules/diff-surfel-rasterization')
25
+ # import simple_knn
26
+ # import diff_surfel_rasterization
27
 
28
  import argparse
29
  import spaces
requirements.txt CHANGED
@@ -44,4 +44,5 @@ onnxscript
44
  tritonclient
45
  gradio==4.43.0
46
  rembg
47
- onnxruntime
 
 
44
  tritonclient
45
  gradio==4.43.0
46
  rembg
47
+ onnxruntime
48
+ git+https://github.com/facebookresearch/pytorch3d.git@stable