YaohuiW commited on
Commit
4039503
·
verified ·
1 Parent(s): e73bbee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -6,6 +6,8 @@ from PIL import Image
6
  import numpy as np
7
  import imageio
8
  import spaces
 
 
9
 
10
  extensions_dir = "./torch_extension/"
11
  os.environ["TORCH_EXTENSIONS_DIR"] = extensions_dir
 
6
  import numpy as np
7
  import imageio
8
  import spaces
9
+ import subprocess
10
+
11
 
12
  extensions_dir = "./torch_extension/"
13
  os.environ["TORCH_EXTENSIONS_DIR"] = extensions_dir