Update inference.py

inference.py

@@ -26,8 +26,7 @@ warnings.filterwarnings('ignore')
 from input_preprocess import *
 
 # Device configuration
-device_idx_ds = 3
-device = torch.device(f'cuda:{device_idx_ds}' if torch.cuda.is_available() else "cpu")
+device = torch.device(cuda if torch.cuda.is_available() else "cpu")
 if torch.cuda.is_available():
     torch.cuda.empty_cache()