test

app.py

@@ -107,15 +107,18 @@ def find_cuda_home():
     except Exception as e:
         print(f"Error finding CUDA_HOME: {e}")
         return None
+    
+
+
+print(torch.cuda.is_available())
+from torch.utils.cpp_extension import _find_cuda_home
+cuda_home = _find_cuda_home()
+if cuda_home is None:
+    raise EnvironmentError("CUDA_HOME could not be found.")
 
 
 @spaces.GPU
 def upsample_features(image, model_option):
-    print(torch.cuda.is_available())
-    from torch.utils.cpp_extension import _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"])