Update app.py
Browse files
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 |
|