Update app.py

app.py

@@ -19,13 +19,11 @@ dummy.__version__ = "0.0.0"
 ops = types.ModuleType("xformers.ops")
 
 def _fake_mea(q, k, v, *_, dropout_p: float = 0.0, **__):
-    # PyTorch 2.x 표준 S-DPA로 대체 (CPU에서도 동작)
     return torch.nn.functional.scaled_dot_product_attention(
         q, k, v, dropout_p=dropout_p, is_causal=False
     )
 
-class _FakeLowerTriangularMask:
-    """audiocraft가 타입 존재만 확인하므로 빈 클래스로 대체"""
+class _FakeLowerTriangularMask:  # audiocraft가 존재 여부만 확인
     pass
 
 ops.memory_efficient_attention = _fake_mea
@@ -35,13 +33,11 @@ sys.modules["xformers"] = dummy
 sys.modules["xformers.ops"] = ops
 
 # ─────────────────────────────────────────────────────────────
-# 2. 기타 누락 모듈에 대한 더미(stub) 안전망
-#    (requirements.txt 에 이미 설치하지만, 혹시 빠져도 런타임 통과)
+# 2. (선택) 설치하지 않은 모듈만 안전망으로 스텁 처리 ★
+#    - 이미 requirements.txt에서 설치한 모듈(librosa, av 등)은
+#      스텁 대상에서 제거합니다.
 # ─────────────────────────────────────────────────────────────
-for name in (
-    "av", "librosa", "torchdiffeq", "torchmetrics",
-    "pesq", "pystoi", "soxr"
-):
+for name in ("pesq", "pystoi", "soxr"):   # ★ 필요시만 남김
     if name not in sys.modules:
         sys.modules[name] = types.ModuleType(name)
 
@@ -52,16 +48,14 @@ try:
     from audiocraft.models import MusicGen
     from audiocraft.data.audio import audio_write
 except ModuleNotFoundError:
-    # 로컬 실행 등으로 미설치 시: 의존성 없는 형태로 설치
     subprocess.check_call([
         sys.executable, "-m", "pip", "install",
         "git+https://github.com/facebookresearch/audiocraft@main",
         "--no-deps", "--use-pep517"
     ])
-    # 필요 최소 의존성만 즉석 설치 (stub로도 대부분 통과하지만 안전하게)
     subprocess.check_call([sys.executable, "-m", "pip", "install",
-                           "encodec", "torchdiffeq", "torchmetrics",
-                           "librosa", "soxr", "av"])
+                           "encodec", "librosa", "av", "torchdiffeq",
+                           "torchmetrics", "num2words"])
     from audiocraft.models import MusicGen
     from audiocraft.data.audio import audio_write
 
@@ -81,10 +75,10 @@ tokenizer = AutoTokenizer.from_pretrained(
 )
 
 # ─────────────────────────────────────────────────────────────
-# 5. MusicGen 모델 (CPU 전용)
+# 5. MusicGen 모델
 # ─────────────────────────────────────────────────────────────
 musicgen = MusicGen.get_pretrained("facebook/musicgen-small")
-musicgen.set_generation_params(duration=10)  # 10초 길이
+musicgen.set_generation_params(duration=10)
 
 # ─────────────────────────────────────────────────────────────
 # 6. 파이프라인 함수