test

app.py

@@ -89,7 +89,9 @@ model_option = gr.Radio(options, value="dino16",
 
 @spaces.GPU
 def upsample_features(image, model_option):
-    print(torch.utils.cpp_extension.CUDA_HOME)
+    from torch.utils.cpp_extension import CUDA_HOME
+
+    print(f"CUDA Home: {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'