test

app.py

@@ -1,6 +1,5 @@
 import os
 import subprocess
-import glob
 
 import matplotlib.pyplot as plt
 import torch
@@ -90,7 +89,7 @@ model_option = gr.Radio(options, value="dino16",
 
 @spaces.GPU
 def upsample_features(image, model_option):
-    print(os.environ['CUDA_HOME'])
+    print(torch.utils.cpp_extension.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'