Docfile commited on
Commit
b8a71c5
1 Parent(s): 313ac00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -27,6 +27,7 @@ from df.enhance import enhance, init_df, load_audio, save_audio
27
  from df.io import resample
28
 
29
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
30
  model, df, _ = init_df("./DeepFilterNet2", config_allow_defaults=True)
31
  model = model.to(device=device).eval()
32
 
 
27
  from df.io import resample
28
 
29
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
30
+ map_location=torch.device('cpu')
31
  model, df, _ = init_df("./DeepFilterNet2", config_allow_defaults=True)
32
  model = model.to(device=device).eval()
33