Jahmori commited on
Commit
19af5b6
·
1 Parent(s): 36bf7b0

swapped back to storage buckets until model repo can be fixed

Browse files
Files changed (1) hide show
  1. app.py +11 -11
app.py CHANGED
@@ -36,7 +36,7 @@ def process_audio(mixture):
36
  Path.mkdir(sources, exist_ok=True)
37
 
38
  # Downloads RE-SepFormer model from huggingface buckets to space local directory
39
- # sync_bucket("hf://buckets/Jahmori-R/PureVoiceAISpace-storage", "./local")
40
 
41
  # ---------------------------------------------------------
42
  # INITIALIZE MODELS
@@ -46,19 +46,19 @@ def process_audio(mixture):
46
 
47
  # Loading Resepformer Model
48
  try:
49
- # separator = SepformerSeparation.from_hparams(
50
- # source=f"./local/resepformer-librimix2spk",
51
- # hparams_file="hyperparams.yaml",
52
- # run_opts={"device": "cuda"},
53
- # )
54
-
55
  separator = SepformerSeparation.from_hparams(
56
- source="Jahmori-R/resepformer-librimix2spk",
57
- savedir=f"./pretrained_models/resepformer-librimix2spk",
58
- # run_opts={"device": "cuda"},
59
- # fetch_config=FetchConfig(token=True)
60
  )
61
 
 
 
 
 
 
 
 
62
  print("RE-SepFormer model successfully loaded")
63
  except Exception as e:
64
  print(f"Failed to load RE-SepFormer model. Error: {e}")
 
36
  Path.mkdir(sources, exist_ok=True)
37
 
38
  # Downloads RE-SepFormer model from huggingface buckets to space local directory
39
+ sync_bucket("hf://buckets/Jahmori-R/PureVoiceAISpace-storage", "./local")
40
 
41
  # ---------------------------------------------------------
42
  # INITIALIZE MODELS
 
46
 
47
  # Loading Resepformer Model
48
  try:
 
 
 
 
 
 
49
  separator = SepformerSeparation.from_hparams(
50
+ source=f"./local/resepformer-librimix2spk",
51
+ hparams_file="hyperparams.yaml",
52
+ run_opts={"device": "cuda"},
 
53
  )
54
 
55
+ # separator = SepformerSeparation.from_hparams(
56
+ # source="Jahmori-R/resepformer-librimix2spk",
57
+ # savedir=f"./pretrained_models/resepformer-librimix2spk",
58
+ # # run_opts={"device": "cuda"},
59
+ # # fetch_config=FetchConfig(token=True)
60
+ # )
61
+
62
  print("RE-SepFormer model successfully loaded")
63
  except Exception as e:
64
  print(f"Failed to load RE-SepFormer model. Error: {e}")