update

app.py

@@ -1,4 +1,3 @@
-import spaces
 import gradio as gr
 import os
 import numpy as np
@@ -22,7 +21,6 @@ def get_md5(content):
     md5 = md5hash.hexdigest()
     return md5
 
-@spaces.GPU(duration=120)
 def get_video_res(img_path, audio_path, res_video_path, dynamic_scale=1.0):
 
     expand_ratio = 0.5

sonic.py

@@ -19,10 +19,12 @@ from src.models.audio_adapter.audio_proj import AudioProjModel
 from src.models.audio_adapter.audio_to_bucket import Audio2bucketModel
 from src.utils.RIFE.RIFE_HDv3 import RIFEModel
 from src.dataset.face_align.align import AlignImage
+import spaces
 
 
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 
+@torch.no_grad()
 def test(
     pipe,
     config,
@@ -254,7 +256,7 @@ class Sonic():
         crop_image = face_image[crop_bbox[1]:crop_bbox[3], crop_bbox[0]:crop_bbox[2]]
         cv2.imwrite(output_image_path, crop_image)
 
-    @torch.no_grad()
+    @spaces.GPU(duration=120)
     def process(self,
                 image_path,
                 audio_path,