tummi76 commited on
Commit
16ce2c8
·
verified ·
1 Parent(s): 3f04617

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import spaces
8
  bundle = torchaudio.pipelines.HDEMUCS_HIGH_MUSDB_PLUS
9
  model = bundle.get_model()
10
  sample_rate = bundle.sample_rate
11
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
12
  model.to(device)
13
 
14
 
 
8
  bundle = torchaudio.pipelines.HDEMUCS_HIGH_MUSDB_PLUS
9
  model = bundle.get_model()
10
  sample_rate = bundle.sample_rate
11
+ device = torch.device("cpu")
12
  model.to(device)
13
 
14