Pecorized commited on
Commit
758f373
·
1 Parent(s): f6e8343

print devices

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -13,8 +13,10 @@ def inference(audio):
13
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
14
  if device=='cuda':
15
  use_cuda=True
 
16
  else:
17
  use_cuda=False
 
18
  try:
19
 
20
  # Using subprocess.run for better control
 
13
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
14
  if device=='cuda':
15
  use_cuda=True
16
+ print(f"Using device: {device}")
17
  else:
18
  use_cuda=False
19
+ print(f"Using device: {device}")
20
  try:
21
 
22
  # Using subprocess.run for better control