vCLR

Running on Zero

allencbzhang commited on Apr 4

Commit

010e2ac

verified ·

1 Parent(s): ef08f4e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,6 +3,7 @@ import os
 import gradio as gr
 import shutil
 def find_cuda():
     # Check if CUDA_HOME or CUDA_PATH environment variables are set
     cuda_home = os.environ.get('CUDA_HOME') or os.environ.get('CUDA_PATH')

 import gradio as gr
 import shutil
+@spaces.GPU
 def find_cuda():
     # Check if CUDA_HOME or CUDA_PATH environment variables are set
     cuda_home = os.environ.get('CUDA_HOME') or os.environ.get('CUDA_PATH')