Update app.py

app.py

@@ -1,25 +1,24 @@
-# ---------------------------------------------------------
-# app.py   (2025-05 rev, aligned with latest sonic.py)
-# ---------------------------------------------------------
-import os, io, hashlib, numpy as np, gradio as gr, spaces
+import os, io, hashlib, gradio as gr, spaces
 from pydub import AudioSegment
 from PIL import Image
 from sonic import Sonic
 
-# ------------------------------------------------------------------
-# 1) 모델 & 체크포인트 다운로드 (최초 1회)
-# ------------------------------------------------------------------
-os.system(
-    'python3 -m pip install "huggingface_hub[cli]" accelerate -q; '
-    'huggingface-cli download LeonJoe13/Sonic '
-    '      --local-dir checkpoints -q; '
-    'huggingface-cli download stabilityai/stable-video-diffusion-img2vid-xt '
-    '      --local-dir checkpoints/stable-video-diffusion-img2vid-xt -q; '
-    'huggingface-cli download openai/whisper-tiny '
-    '      --local-dir checkpoints/whisper-tiny -q'
-)
-
-pipe = Sonic()      # GPU 초기화
+# --------------------------------------------------------------
+# 1) 필요 체크포인트(두 개)만 받는다
+# --------------------------------------------------------------
+DL_CMDS = [
+    # Sonic(가중치, unet 등)
+    "huggingface-cli download LeonJoe13/Sonic --local-dir checkpoints/LeonJoe13-Sonic --local-dir-use-symlinks False -q",
+    # stable-video-diffusion-img2vid-xt (VAE/UNet/CLIP)
+    "huggingface-cli download stabilityai/stable-video-diffusion-img2vid-xt "
+    "--local-dir checkpoints/stable-video-diffusion-img2vid-xt --local-dir-use-symlinks False -q",
+    # whisper-tiny
+    "huggingface-cli download openai/whisper-tiny --local-dir checkpoints/whisper-tiny --local-dir-use-symlinks False -q",
+]
+for cmd in DL_CMDS:
+    os.system(cmd)
+
+pipe = Sonic()   # 위에서 모델 경로를 자동으로 찾음
 
 # ------------------------------------------------------------------
 def md5(b: bytes) -> str:          # 빠른 32-byte 해시