update app.py

app.py

@@ -1,6 +1,8 @@
 
 import subprocess, os, sys
 
+os.environ["CUDA_VISIBLE_DEVICES"] = "0"
+
 sys.path.insert(0, './GroundingDINO')
 
 result = subprocess.run(['pip', 'list'], check=True)
@@ -44,7 +46,7 @@ from huggingface_hub import hf_hub_download
 def get_device():
     from numba import cuda
     if cuda.is_available():
-        device = cuda.get_current_device()
+        device = 'cuda:0' # cuda.get_current_device()
     else:
         device = 'cpu'
     return device