root commited on
Commit
6d09ae4
·
1 Parent(s): efee030
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -16,14 +16,14 @@ from download import download_model
16
  # 下载模型
17
  APP_DIR = op.dirname(op.abspath(__file__))
18
  download_model(APP_DIR)
19
- # base_full_path = op.join(APP_DIR, "ckpt", "songgeneration_base_full")
20
- # os.makedirs(base_full_path, exist_ok=True)
21
  download_model(op.join(APP_DIR, "ckpt"), repo_id="lglg666/SongGeneration-large", revision="75e2043")
22
  print("Successful downloaded model.")
23
 
24
  # 模型初始化
25
  from levo_inference import LeVoInference
26
- MODEL = LeVoInference(op.join(APP_DIR, "ckpt", "SongGeneration-large"))
27
 
28
  EXAMPLE_LYRICS = """
29
  [intro-medium]
 
16
  # 下载模型
17
  APP_DIR = op.dirname(op.abspath(__file__))
18
  download_model(APP_DIR)
19
+ base_full_path = op.join(APP_DIR, "ckpt", "songgeneration_large")
20
+ os.makedirs(base_full_path, exist_ok=True)
21
  download_model(op.join(APP_DIR, "ckpt"), repo_id="lglg666/SongGeneration-large", revision="75e2043")
22
  print("Successful downloaded model.")
23
 
24
  # 模型初始化
25
  from levo_inference import LeVoInference
26
+ MODEL = LeVoInference(base_full_path)
27
 
28
  EXAMPLE_LYRICS = """
29
  [intro-medium]