Spaces:
dylanebert
/
Running on Zero

dylanebert HF staff commited on
Commit
0a8057f
1 Parent(s): b7c04cb

install diff_gaussian_rasterization on zerogpu

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -9,7 +9,12 @@ from gradio_client import Client, file
9
  try:
10
  import diff_gaussian_rasterization # noqa: F401
11
  except ImportError:
12
- os.system("pip install ./diff-gaussian-rasterization")
 
 
 
 
 
13
 
14
  TMP_DIR = "/tmp"
15
  os.makedirs(TMP_DIR, exist_ok=True)
 
9
  try:
10
  import diff_gaussian_rasterization # noqa: F401
11
  except ImportError:
12
+
13
+ @spaces.GPU
14
+ def install_diff_gaussian_rasterization():
15
+ os.system("pip install ./diff-gaussian-rasterization")
16
+
17
+ install_diff_gaussian_rasterization()
18
 
19
  TMP_DIR = "/tmp"
20
  os.makedirs(TMP_DIR, exist_ok=True)