Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
6d804ea
1
Parent(s):
69076fe
Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ css = """
|
|
| 22 |
height: 62px;
|
| 23 |
}
|
| 24 |
"""
|
| 25 |
-
|
| 26 |
model_configs = {
|
| 27 |
'vits': {'encoder': 'vits', 'features': 64, 'out_channels': [48, 96, 192, 384]},
|
| 28 |
'vitb': {'encoder': 'vitb', 'features': 128, 'out_channels': [96, 192, 384, 768]},
|
|
|
|
| 22 |
height: 62px;
|
| 23 |
}
|
| 24 |
"""
|
| 25 |
+
DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 26 |
model_configs = {
|
| 27 |
'vits': {'encoder': 'vits', 'features': 64, 'out_channels': [48, 96, 192, 384]},
|
| 28 |
'vitb': {'encoder': 'vitb', 'features': 128, 'out_channels': [96, 192, 384, 768]},
|