Upload gradio_demo.py

src/gradio_demo.py

@@ -1,3 +1,4 @@
+import spaces
 import torch, uuid
 import os, sys, shutil
 from src.utils.preprocess import CropAndExtract
@@ -33,6 +34,7 @@ class SadTalker():
         self.config_path = config_path
       
 
+    @spaces.GPU()
     def test(self, source_image, driven_audio, preprocess='crop', 
         still_mode=False,  use_enhancer=False, batch_size=1, size=256, 
         pose_style = 0, exp_scale=1.0,