test

app.py

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