Update app.py

app.py

@@ -232,8 +232,6 @@ def get_last_mp3_file(output_dir):
     mp3_files_with_path.sort(key=os.path.getmtime, reverse=True)
     return mp3_files_with_path[0]
 
-
-
 def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
     try:
         # 모델 선택 및 설정
@@ -255,7 +253,7 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
         os.makedirs(output_dir, exist_ok=True)
         empty_output_folder(output_dir)
 
-        # 기본 명령어 구성
+        # 기본 명령어 구성 (불필요한 매개변수 제거)
         command = [
             "python", "infer.py",
             "--stage1_model", model_path,
@@ -266,16 +264,12 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
             "--stage2_batch_size", str(config['batch_size']),
             "--output_dir", output_dir,
             "--cuda_idx", "0",
-            "--max_new_tokens", str(actual_max_tokens),
-            "--temperature", str(config['temperature']),
-            "--chorus_strength", str(config['chorus_strength'])
+            "--max_new_tokens", str(actual_max_tokens)
         ]
 
         # GPU가 있는 경우에만 추가 옵션 적용
         if torch.cuda.is_available():
-            command.extend([
-                "--disable_offload_model"
-            ])
+            command.append("--disable_offload_model")
 
         # CUDA 환경 변수 설정
         env = os.environ.copy()
@@ -288,6 +282,13 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
                 "PYTORCH_CUDA_ALLOC_CONF": "max_split_size_mb:512"
             })
 
+        # transformers 캐시 마이그레이션 처리
+        try:
+            from transformers.utils import move_cache
+            move_cache()
+        except Exception as e:
+            logging.warning(f"Cache migration warning (non-critical): {e}")
+
         # 명령 실행
         process = subprocess.run(
             command,
@@ -333,16 +334,7 @@ def main():
     with gr.Blocks() as demo:
         with gr.Column():
             gr.Markdown("# YuE: Open Music Foundation Models for Full-Song Generation (Multi-Language Support)")
-            gr.HTML("""
-            <div style="display:flex;column-gap:4px;">
-                <a href="https://github.com/multimodal-art-projection/YuE">
-                    <img src='https://img.shields.io/badge/GitHub-Repo-blue'>
-                </a> 
-                <a href="https://map-yue.github.io">
-                    <img src='https://img.shields.io/badge/Project-Page-green'>
-                </a>
-            </div>
-            """)
+
             
             with gr.Row():
                 with gr.Column():
@@ -394,17 +386,6 @@ With you here beside me, everything's alright
 Can't imagine life alone, don't want to let you go
 Stay with me forever, let our love just flow
 
-[verse]
-Morning light is breaking, through the window pane
-Memories of yesterday, like soft summer rain
-In your arms I'm finding, all I'm dreaming of
-Every day beside you, fills my heart with love
-
-[chorus]
-Don't let this moment fade, hold me close tonight
-With you here beside me, everything's alright
-Can't imagine life alone, don't want to let you go
-Stay with me forever, let our love just flow
                         """
                     ],
                     # 한국어 예제
@@ -422,17 +403,6 @@ Stay with me forever, let our love just flow
 영원히 계속될 우리의 노래
 이 순간을 기억해 forever
 
-[verse]
-새로운 내일을 향해 나아가
-우리만의 길을 만들어가
-믿음으로 가득한 우리의 맘
-절대 멈추지 않아 계속해서
-
-[chorus]
-달려가자 더 높이 더 멀리
-두려움은 없어 너와 함께라면
-영원히 계속될 우리의 노래
-이 순간을 기억해 forever
                         """
                     ]
                 ],