Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,18 +11,17 @@ import time
|
|
| 11 |
from huggingface_hub import snapshot_download
|
| 12 |
|
| 13 |
repo_id = "amphion/anyaccomp"
|
| 14 |
-
|
| 15 |
|
| 16 |
print(f"Downloading model files from {repo_id}...")
|
| 17 |
-
model_dir = snapshot_download(repo_id=repo_id, local_dir=
|
| 18 |
print(f"Model files downloaded to: {model_dir}")
|
| 19 |
|
| 20 |
-
CFG_PATH = os.path.join(
|
|
|
|
| 21 |
CHECKPOINT_PATH = os.path.join(model_dir, "pretrained/flow_matching")
|
| 22 |
VOCODER_CHECKPOINT_PATH = os.path.join(model_dir, "pretrained/vocoder")
|
| 23 |
-
VOCODER_CFG_PATH = os.path.join(model_dir, "config/vocoder.json")
|
| 24 |
|
| 25 |
-
base_dir = os.path.dirname(os.path.abspath(__file__))
|
| 26 |
INFER_DST = os.path.join(base_dir, "output_gradio")
|
| 27 |
EXAMPLE_DIR = os.path.join(base_dir, "example/gradio")
|
| 28 |
|
|
|
|
| 11 |
from huggingface_hub import snapshot_download
|
| 12 |
|
| 13 |
repo_id = "amphion/anyaccomp"
|
| 14 |
+
base_dir = os.path.dirname(os.path.abspath(__file__))
|
| 15 |
|
| 16 |
print(f"Downloading model files from {repo_id}...")
|
| 17 |
+
model_dir = snapshot_download(repo_id=repo_id, local_dir=base_dir)
|
| 18 |
print(f"Model files downloaded to: {model_dir}")
|
| 19 |
|
| 20 |
+
CFG_PATH = os.path.join(base_dir, "config/flow_matching.json")
|
| 21 |
+
VOCODER_CFG_PATH = os.path.join(base_dir, "config/vocoder.json")
|
| 22 |
CHECKPOINT_PATH = os.path.join(model_dir, "pretrained/flow_matching")
|
| 23 |
VOCODER_CHECKPOINT_PATH = os.path.join(model_dir, "pretrained/vocoder")
|
|
|
|
| 24 |
|
|
|
|
| 25 |
INFER_DST = os.path.join(base_dir, "output_gradio")
|
| 26 |
EXAMPLE_DIR = os.path.join(base_dir, "example/gradio")
|
| 27 |
|