Spaces:
Running
on
L40S
Running
on
L40S
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 |
|