update

app.py

@@ -6,11 +6,10 @@ from pydub import AudioSegment
 import hashlib
 from sonic import Sonic
 
-# cmd = 'python3 -m pip install "huggingface_hub[cli]"; \
-# huggingface-cli download LeonJoe13/Sonic --local-dir  checkpoints; \
-# huggingface-cli download stabilityai/stable-video-diffusion-img2vid-xt --local-dir  checkpoints/stable-video-diffusion-img2vid-xt; \
-# huggingface-cli download openai/whisper-tiny --local-dir checkpoints/whisper-tiny; \
-cmd = 'nvidia-smi'
+cmd = 'python3 -m pip install "huggingface_hub[cli]"; \
+huggingface-cli download LeonJoe13/Sonic --local-dir  checkpoints; \
+huggingface-cli download stabilityai/stable-video-diffusion-img2vid-xt --local-dir  checkpoints/stable-video-diffusion-img2vid-xt; \
+huggingface-cli download openai/whisper-tiny --local-dir checkpoints/whisper-tiny;'
 os.system(cmd)
 
 import torch
@@ -23,7 +22,6 @@ def get_md5(content):
     md5 = md5hash.hexdigest()
     return md5
 
-@spaces.GPU
 def get_video_res(img_path, audio_path, res_video_path, dynamic_scale=1.0):
 
     expand_ratio = 0.5
@@ -45,6 +43,7 @@ res_path = './res_path/'
 os.makedirs(tmp_path,exist_ok=1)
 os.makedirs(res_path,exist_ok=1)
 
+@spaces.GPU
 def process_sonic(image,audio,s0):
     img_md5= get_md5(np.array(image))
     audio_md5 = get_md5(audio[1])