Spaces:
Running on Zero
Running on Zero
Jahmori commited on
Commit ·
19af5b6
1
Parent(s): 36bf7b0
swapped back to storage buckets until model repo can be fixed
Browse files
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 |
-
|
| 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="
|
| 57 |
-
|
| 58 |
-
|
| 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}")
|