BorisovMaksim commited on
Commit
8dd2b87
·
1 Parent(s): 685fbad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ import torch
13
 
14
  @hydra.main(version_base=None, config_path="conf", config_name="config")
15
  def run_app(cfg: DictConfig):
16
- model = Demucs(cfg['model'])
17
  model_path = hf_hub_download(repo_id="BorisovMaksim/demucs", filename="Demucs_original_sr_epoch3.pt")
18
  checkpoint = torch.load(model_path)
19
  model.load_state_dict(checkpoint['model_state_dict'])
 
13
 
14
  @hydra.main(version_base=None, config_path="conf", config_name="config")
15
  def run_app(cfg: DictConfig):
16
+ model = Demucs(cfg['model']['demucs'])
17
  model_path = hf_hub_download(repo_id="BorisovMaksim/demucs", filename="Demucs_original_sr_epoch3.pt")
18
  checkpoint = torch.load(model_path)
19
  model.load_state_dict(checkpoint['model_state_dict'])