change ckpt name & UI

app.py

@@ -14,7 +14,7 @@ print("Successful downloaded model.")
 from levo_inference import LeVoInference
 
 # 模型初始化
-MODEL = LeVoInference(op.join(APP_DIR, "ckpt/songgeneration_base_zh/"))
+MODEL = LeVoInference(op.join(APP_DIR, "ckpt/songgeneration_base/"))
 
 EXAMPLE_LYRICS = """
 [intro-short]
@@ -154,11 +154,11 @@ lyrics
             with gr.Tabs(elem_id="extra-tabs"):
                 with gr.Tab("Genre Select"):
                     genre = gr.Radio(
-                        choices=["Auto", "Pop", "R&B", "Dance", "Jazz", "Folk", "Rock", "Chinese Style", "Chinese Tradition", "Metal", "Reggae", "Chinese Opera"],
+                        choices=["Pop", "R&B", "Dance", "Jazz", "Folk", "Rock", "Chinese Style", "Chinese Tradition", "Metal", "Reggae", "Chinese Opera", "Auto"],
                         label="Genre Select(Optional)",
-                        value="Auto",  # 默认选中第一个
+                        value="Pop",
                         interactive=True,
-                        elem_id="single-select-radio"  # 便于自定义样式
+                        elem_id="single-select-radio"
                     )
                 with gr.Tab("Audio Prompt"):
                     prompt_audio = gr.Audio(
@@ -167,9 +167,10 @@ lyrics
                         elem_id="audio-prompt"
                     )
                 with gr.Tab("Text Prompt"):
+                    gr.Markdown("For detailed usage, please refer to [here](https://github.com/tencent-ailab/SongGeneration?tab=readme-ov-file#-description-input-format)")
                     description = gr.Textbox(
                         label="Song Description (Optional)",
-                        info="Describe the gender, timbre, genre, emotion, instrument and bpm of the song",
+                        info="Describe the gender, timbre, genre, emotion, instrument and bpm of the song. Only English is supported currently.​",
                         placeholder="female, dark, pop, sad, piano and drums, the bpm is 125.",
                         lines=1,
                         max_lines=2

download.py

@@ -7,7 +7,7 @@ def download_model(local_dir):
     downloaded_path = snapshot_download(
         repo_id=repo_id,
         local_dir=local_dir,
-        revision="de30f7b",
+        revision="0c80d30",
         token=os.environ.get("HF_TOKEN"), 
         ignore_patterns=['.git*']
     )