zero gpu ready

gradio_app.py

@@ -41,6 +41,7 @@ from dataset_process import audio
 
 from dataset_process.croper import Croper
 
+import spaces
 
 def parse_audio_length(audio_length, sr, fps):
     bit_per_frames = sr / fps
@@ -376,6 +377,7 @@ class Inferencer(object):
         os.remove(path)
         os.remove(new_audio_path)
 
+@spaces.GPU()
 def gradio_infer(source_image, driven_audio):
     
     import tempfile