test

app.py

@@ -109,7 +109,8 @@ def find_cuda_home():
 
 @spaces.GPU
 def upsample_features(image, model_option):
-    os.environ["CUDA_HOME"] = find_cuda_home()
+    from torch.utils.cpp_extension import _find_cuda_home
+    os.environ["CUDA_HOME"] = _find_cuda_home()
     print(os.environ["CUDA_HOME"])
     os.environ['PATH'] = '/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin'
     os.environ['LD_LIBRARY_PATH'] = '/usr/local/nvidia/lib:/usr/local/nvidia/lib64'