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