Update app.py

app.py

@@ -7,8 +7,10 @@ import torch
 import logging
 import numpy as np
 import re
+import sys  # sys 모듈 추가
 from concurrent.futures import ThreadPoolExecutor
 from functools import lru_cache
+from datetime import datetime
 
 # 로깅 설정
 logging.basicConfig(
@@ -324,7 +326,6 @@ def get_audio_duration(file_path):
         logging.error(f"Failed to get audio duration: {e}")
         return None
 
-
 def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
     genre_txt_path = None
     lyrics_txt_path = None
@@ -336,7 +337,6 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
         model_path, config, params = optimize_model_selection(lyrics_txt_content, genre_txt_content)
         logging.info(f"Selected model: {model_path}")
         logging.info(f"Lyrics analysis: {params}")
-        
 
         
         has_chorus = params['sections']['chorus'] > 0
@@ -365,9 +365,10 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
         empty_output_folder(output_dir)
 
 
+        python_executable = sys.executable or "python"  # fallback to "python" if sys.executable is not available
         
         command = [
-            sys.executable,
+            python_executable,
             INFER_SCRIPT,
             "--stage1_model", model_path,
             "--stage2_model", "m-a-p/YuE-s2-1B-general",
@@ -380,7 +381,6 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
             "--max_new_tokens", str(actual_max_tokens),
             "--disable_offload_model"
         ]
-        
 
 
         env = os.environ.copy()