Ahsen Khaliq commited on
Commit
0df1e05
1 Parent(s): 85339f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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