waytan22 commited on
Commit
83e833c
·
1 Parent(s): d658154
Files changed (1) hide show
  1. app.py +2 -3
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