test

app.py

@@ -10,6 +10,7 @@ import os
 import requests
 import csv
 import spaces
+from triton.fb import build_paths
 
 
 def plot_feats(image, lr, hr):
@@ -108,10 +109,8 @@ def find_cuda_home():
 
 @spaces.GPU
 def upsample_features(image, model_option):
-    cuda_home = find_cuda_home()
-    print(cuda_home)
-
-    os.environ['CUDA_HOME'] = cuda_home
+    os.environ["CUDA_HOME"] = build_paths.cuda()
+    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'