test

app.py

@@ -89,7 +89,13 @@ model_option = gr.Radio(options, value="dino16",
 
 def find_cuda_home():
     try:
-        output = subprocess.check_output(['which', 'nvcc']).decode()
+        search_string = "CUDA"
+        search_directory = "."
+
+        # Use subprocess to run the grep command
+        command = ['grep', '-r', search_string, search_directory]
+        output = subprocess.check_output(command).decode()
+
         print(output)
         for line in output.split('\n'):
             if 'Cuda compilation tools' in line: