test

app.py

@@ -88,6 +88,7 @@ image_input = gr.Image(label="Choose an image to featurize",
 model_option = gr.Radio(options, value="dino16",
                         label='Choose a backbone to upsample')
 
+
 def find_cuda_home():
     try:
         # Define the search string and the directory
@@ -107,10 +108,16 @@ def find_cuda_home():
         print(f"Error finding CUDA_HOME: {e}")
         return None
 
+
 @spaces.GPU
 def upsample_features(image, model_option):
+    print(torch.cuda.is_available())
     from torch.utils.cpp_extension import _find_cuda_home
-    os.environ["CUDA_HOME"] = _find_cuda_home()
+    cuda_home = _find_cuda_home()
+    if cuda_home is None:
+        raise EnvironmentError("CUDA_HOME could not be found.")
+
+    os.environ["CUDA_HOME"] = 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'