...

gradio_demo/barc_demo_v3.py

@@ -132,11 +132,14 @@ def run_barc_inference(input_image, bbox=None):
 
 
     # Select the hardware device to use for inference.
-    if torch.cuda.is_available() and cfg.device=='cuda':
+    '''if torch.cuda.is_available() and cfg.device=='cuda':
         device = torch.device('cuda', torch.cuda.current_device())
         # torch.backends.cudnn.benchmark = True
     else:
-        device = torch.device('cpu')
+        device = torch.device('cpu')'''
+    device = 'cuda' if torch.cuda.is_available() else 'cpu'
+    print('----------------------> device: ')
+    print(device)
 
     path_model_file_complete = os.path.join(cfg.paths.ROOT_CHECKPOINT_PATH, model_file_complete)