Update app.py

app.py

@@ -1,9 +1,14 @@
+# 1. 먼저 로깅 설정
+import logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# 2. 나머지 imports
 import os
 import time
 from datetime import datetime
 import gradio as gr
 import torch
-import logging
 import requests
 from pathlib import Path
 import cv2
@@ -26,9 +31,15 @@ from mmaudio.model.networks import MMAudio, get_my_mmaudio
 from mmaudio.model.sequence_config import SequenceConfig
 from mmaudio.model.utils.features_utils import FeaturesUtils
 
+# 3. API 설정
+CATBOX_USER_HASH = "30f52c895fd9d9cb387eee489"
+REPLICATE_API_TOKEN = os.getenv("API_KEY")
 
+# 4. 오디오 모델 설정
+device = 'cuda'
+dtype = torch.bfloat16
 
-# 먼저 get_model 함수 정의
+# 5. get_model 함수 정의
 def get_model() -> tuple[MMAudio, FeaturesUtils, SequenceConfig]:
     seq_cfg = model.seq_cfg
 
@@ -46,10 +57,7 @@ def get_model() -> tuple[MMAudio, FeaturesUtils, SequenceConfig]:
 
     return net, feature_utils, seq_cfg
 
-# 그 다음 모델 설정 및 초기화
-device = 'cuda'
-dtype = torch.bfloat16
-
+# 6. 모델 초기화
 model: ModelConfig = all_model_cfg['large_44k_v2']
 model.download_if_needed()
 output_dir = Path('./output/gradio')
@@ -58,15 +66,6 @@ setup_eval_logging()
 net, feature_utils, seq_cfg = get_model()
 
 
-# 로깅 설정
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-# API 설정
-CATBOX_USER_HASH = "30f52c895fd9d9cb387eee489"
-REPLICATE_API_TOKEN = os.getenv("API_KEY")
-
-
 
 @spaces.GPU(duration=120)
 @torch.inference_mode()