Spaces:
Build error
Build error
root
commited on
Commit
·
83e833c
1
Parent(s):
d658154
fix bug
Browse files
app.py
CHANGED
|
@@ -5,14 +5,14 @@ import yaml
|
|
| 5 |
import time
|
| 6 |
import os.path as op
|
| 7 |
from download import download_model
|
| 8 |
-
from levo_inference import LeVoInference
|
| 9 |
-
|
| 10 |
|
| 11 |
# 下载模型
|
| 12 |
APP_DIR = op.dirname(op.abspath(__file__))
|
| 13 |
download_model(APP_DIR)
|
| 14 |
print("Successful downloaded model.")
|
| 15 |
|
|
|
|
|
|
|
| 16 |
# 模型初始化
|
| 17 |
MODEL = LeVoInference(op.join(APP_DIR, "ckpt/songgeneration_base_zn/"))
|
| 18 |
|
|
@@ -100,7 +100,6 @@ def generate_song(lyric, description=None, prompt_audio=None, genre=None, cfg_co
|
|
| 100 |
"inference_duration": end - start,
|
| 101 |
"timestamp": datetime.now().isoformat(),
|
| 102 |
}
|
| 103 |
-
print(input_config)
|
| 104 |
|
| 105 |
return (sample_rate, audio_data), json.dumps(input_config, indent=2)
|
| 106 |
|
|
|
|
| 5 |
import time
|
| 6 |
import os.path as op
|
| 7 |
from download import download_model
|
|
|
|
|
|
|
| 8 |
|
| 9 |
# 下载模型
|
| 10 |
APP_DIR = op.dirname(op.abspath(__file__))
|
| 11 |
download_model(APP_DIR)
|
| 12 |
print("Successful downloaded model.")
|
| 13 |
|
| 14 |
+
from levo_inference import LeVoInference
|
| 15 |
+
|
| 16 |
# 模型初始化
|
| 17 |
MODEL = LeVoInference(op.join(APP_DIR, "ckpt/songgeneration_base_zn/"))
|
| 18 |
|
|
|
|
| 100 |
"inference_duration": end - start,
|
| 101 |
"timestamp": datetime.now().isoformat(),
|
| 102 |
}
|
|
|
|
| 103 |
|
| 104 |
return (sample_rate, audio_data), json.dumps(input_config, indent=2)
|
| 105 |
|