Update app.py

app.py

@@ -199,78 +199,36 @@ def install_flash_attn():
         logging.warning(f"Failed to install flash-attn: {e}")
         return False
 
+
 def initialize_system():
     optimize_gpu_settings()
     
-    try:
-        # 기본 디렉토리 구조 생성
-        base_dir = os.path.abspath("./inference")
-        os.makedirs(base_dir, exist_ok=True)
-        os.makedirs(os.path.join(base_dir, "models"), exist_ok=True)
-        
-        # 작업 디렉토리 변경
-        os.chdir(base_dir)
-        logging.info(f"Working directory changed to: {os.getcwd()}")
+    with ThreadPoolExecutor(max_workers=4) as executor:
+        futures = []
         
-        from huggingface_hub import snapshot_download, hf_hub_download
+        futures.append(executor.submit(install_flash_attn))
         
-        # xcodec_mini_infer 다운로드
-        xcodec_dir = "xcodec_mini_infer"
-        os.makedirs(xcodec_dir, exist_ok=True)
+        from huggingface_hub import snapshot_download
         
-        # 필수 파일 다운로드
-        snapshot_download(
-            repo_id="m-a-p/xcodec_mini_infer",
-            local_dir=xcodec_dir,
-            force_download=True
-        )
-        
-        # infer.py 파일 다운로드
-        infer_script = hf_hub_download(
+        folder_path = './inference/xcodec_mini_infer'
+        os.makedirs(folder_path, exist_ok=True)
+        logging.info(f"Created folder at: {folder_path}")
+
+        futures.append(executor.submit(
+            snapshot_download,
             repo_id="m-a-p/xcodec_mini_infer",
-            filename="infer.py",
-            force_download=True
-        )
-        
-        # infer.py 파일을 현재 디렉토리로 복사
-        shutil.copy2(infer_script, "infer.py")
-        
-        # YuE 모델들 다운로드
-        models = [
-            "m-a-p/YuE-s1-7B-anneal-jp-kr-cot",
-            "m-a-p/YuE-s1-7B-anneal-en-cot",
-            "m-a-p/YuE-s1-7B-anneal-zh-cot",
-            "m-a-p/YuE-s2-1B-general"
-        ]
-        
-        for model in models:
-            model_name = model.split('/')[-1]
-            model_path = os.path.join("models", model_name)
-            snapshot_download(
-                repo_id=model,
-                local_dir=model_path,
-                force_download=True
-            )
-        
-        # 필요한 파일들 존재 확인
-        required_files = [
-            "infer.py",
-            os.path.join(xcodec_dir, "config.json"),
-            os.path.join(xcodec_dir, "vocal_decoder.pth"),
-            os.path.join(xcodec_dir, "inst_decoder.pth")
-        ]
-        
-        for file_path in required_files:
-            if not os.path.exists(file_path):
-                raise FileNotFoundError(f"Required file not found: {file_path}")
-            else:
-                file_size = os.path.getsize(file_path)
-                logging.info(f"Verified {file_path}: {file_size} bytes")
+            local_dir="./inference/xcodec_mini_infer",
+            resume_download=True
+        ))
         
-        logging.info("System initialization completed successfully")
-        
-    except Exception as e:
-        logging.error(f"Initialization error: {e}")
+        for future in futures:
+            future.result()
+
+    try:
+        os.chdir("./inference")
+        logging.info(f"Working directory changed to: {os.getcwd()}")
+    except FileNotFoundError as e:
+        logging.error(f"Directory error: {e}")
         raise
 
 @lru_cache(maxsize=100)