viewfinder-annn commited on
Commit
d231354
·
verified ·
1 Parent(s): 28aa1a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -11,18 +11,17 @@ import time
11
  from huggingface_hub import snapshot_download
12
 
13
  repo_id = "amphion/anyaccomp"
14
- local_dir = "anyaccomp_model"
15
 
16
  print(f"Downloading model files from {repo_id}...")
17
- model_dir = snapshot_download(repo_id=repo_id, local_dir=local_dir)
18
  print(f"Model files downloaded to: {model_dir}")
19
 
20
- CFG_PATH = os.path.join(model_dir, "config/flow_matching.json")
 
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