Update app.py

app.py

@@ -306,22 +306,21 @@ 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 - 지원되지 않는 인수 제거
         command = [
-            "python", "./infer.py",  # 상대 경로로 수정
+            "python", "infer.py",
             "--stage1_model", model_path,
             "--stage2_model", "m-a-p/YuE-s2-1B-general",
             "--genre_txt", genre_txt_path,
             "--lyrics_txt", lyrics_txt_path,
             "--run_n_segments", str(actual_num_segments),
             "--stage2_batch_size", "16",
-            "--output_dir", "./output",
+            "--output_dir", output_dir,
             "--cuda_idx", "0",
             "--max_new_tokens", str(actual_max_tokens),
-            "--disable_offload_model"
+            "--disable_offload_model"  # GPU 메모리 최적화를 위해 추가
         ]
 
-
-
         env = os.environ.copy()
         if torch.cuda.is_available():
             env.update({