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
|