Spaces:

BorisovMaksim
/

denoising

Runtime error

BorisovMaksim commited on May 21, 2023

Commit

99682d1

1 Parent(s): 8dd2b87

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ import torch
 def run_app(cfg: DictConfig):
     model = Demucs(cfg['model']['demucs'])
     model_path = hf_hub_download(repo_id="BorisovMaksim/demucs", filename="Demucs_original_sr_epoch3.pt")
-    checkpoint = torch.load(model_path)
     model.load_state_dict(checkpoint['model_state_dict'])
     def denoising_transform(audio):

 def run_app(cfg: DictConfig):
     model = Demucs(cfg['model']['demucs'])
     model_path = hf_hub_download(repo_id="BorisovMaksim/demucs", filename="Demucs_original_sr_epoch3.pt")
+    checkpoint = torch.load(model_path, map_location=torch.device('cpu'))
     model.load_state_dict(checkpoint['model_state_dict'])
     def denoising_transform(audio):