Spaces:
Running
on
L40S
Running
on
L40S
root
commited on
Commit
·
6d09ae4
1
Parent(s):
efee030
fix bug
Browse files
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 |
-
|
| 20 |
-
|
| 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(
|
| 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]
|