Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
•
0df1e05
1
Parent(s):
85339f4
Update app.py
Browse files
app.py
CHANGED
@@ -17,6 +17,7 @@ from feature_extraction.demo_utils import (calculate_codebook_bitrate,
|
|
17 |
from sample_visualization import tensor_to_plt
|
18 |
from torch.utils.data.dataloader import default_collate
|
19 |
|
|
|
20 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
21 |
|
22 |
|
|
|
17 |
from sample_visualization import tensor_to_plt
|
18 |
from torch.utils.data.dataloader import default_collate
|
19 |
|
20 |
+
os.chdir("SpecVQGAN")
|
21 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
22 |
|
23 |
|