Update app.py

app.py

@@ -13,52 +13,55 @@ print("Successful downloaded model.")
 
 from levo_inference import LeVoInference
 
-# 模型初始化
-MODEL = LeVoInference(op.join(APP_DIR, "ckpt/songgeneration_base/"))
-
-EXAMPLE_LYRICS = """
-[intro-short]
-
-[verse]
-夜晚的街灯闪烁
-我漫步在熟悉的角落
-回忆像潮水般涌来
-你的笑容如此清晰
-在心头无法抹去
-那些曾经的甜蜜
-如今只剩我独自回忆
-
-[verse]
-手机屏幕亮起
-是你发来的消息
-简单的几个字
-却让我泪流满面
-曾经的拥抱温暖
-如今却变得遥远
-我多想回到从前
-重新拥有你的陪伴
-
-[chorus]
-回忆的温度还在
-你却已不在
-我的心被爱填满
-却又被思念刺痛
-音乐的节奏奏响
-我的心却在流浪
-没有你的日子
-我该如何继续向前
-
-[outro-short]
-""".strip()
-
-with open(op.join(APP_DIR, 'conf/vocab.yaml'), 'r', encoding='utf-8') as file:
-    STRUCTS = yaml.safe_load(file)
 
 
 # 模拟歌曲生成函数
+@spaces.GPU(duration=100)
 def generate_song(lyric, description=None, prompt_audio=None, genre=None, cfg_coef=None, temperature=None, top_k=None, gen_type="all", progress=gr.Progress(track_tqdm=True)):
-    global MODEL
-    global STRUCTS
+    # 模型初始化
+    MODEL = LeVoInference(op.join(APP_DIR, "ckpt/songgeneration_base/"))
+    
+    EXAMPLE_LYRICS = """
+    [intro-short]
+    
+    [verse]
+    夜晚的街灯闪烁
+    我漫步在熟悉的角落
+    回忆像潮水般涌来
+    你的笑容如此清晰
+    在心头无法抹去
+    那些曾经的甜蜜
+    如今只剩我独自回忆
+    
+    [verse]
+    手机屏幕亮起
+    是你发来的消息
+    简单的几个字
+    却让我泪流满面
+    曾经的拥抱温暖
+    如今却变得遥远
+    我多想回到从前
+    重新拥有你的陪伴
+    
+    [chorus]
+    回忆的温度还在
+    你却已不在
+    我的心被爱填满
+    却又被思念刺痛
+    音乐的节奏奏响
+    我的心却在流浪
+    没有你的日子
+    我该如何继续向前
+    
+    [outro-short]
+    """.strip()
+    
+    with open(op.join(APP_DIR, 'conf/vocab.yaml'), 'r', encoding='utf-8') as file:
+        STRUCTS = yaml.safe_load(file)
+
+    
+    # global MODEL
+    # global STRUCTS
     params = {'cfg_coef':cfg_coef, 'temperature':temperature, 'top_k':top_k}
     params = {k:v for k,v in params.items() if v is not None}
     vocal_structs = ['[verse]', '[chorus]', '[bridge]']