Spaces:
Build error
Build error
print devices
Browse files
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
|