zejunyang commited on
Commit
d1af78b
1 Parent(s): 1061388
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -98,8 +98,10 @@ vis = FaceMeshVisualizer()
98
 
99
  frame_inter_model = init_frame_interpolation_model()
100
 
101
- @spaces.GPU(duration=100)
102
  def audio2video(input_audio, ref_img, headpose_video=None, size=512, steps=25, length=60, seed=42):
 
 
103
  fps = 30
104
  cfg = 3.5
105
  fi_step = 3
@@ -209,8 +211,10 @@ def audio2video(input_audio, ref_img, headpose_video=None, size=512, steps=25, l
209
 
210
  return save_path.replace('_noaudio.mp4', '.mp4'), ref_image_pil
211
 
212
- @spaces.GPU(duration=100)
213
  def video2video(ref_img, source_video, size=512, steps=25, length=60, seed=42):
 
 
214
  cfg = 3.5
215
  fi_step = 3
216
 
 
98
 
99
  frame_inter_model = init_frame_interpolation_model()
100
 
101
+ @spaces.GPU # (duration=100)
102
  def audio2video(input_audio, ref_img, headpose_video=None, size=512, steps=25, length=60, seed=42):
103
+ print('=====Start processing======')
104
+
105
  fps = 30
106
  cfg = 3.5
107
  fi_step = 3
 
211
 
212
  return save_path.replace('_noaudio.mp4', '.mp4'), ref_image_pil
213
 
214
+ @spaces.GPU # (duration=100)
215
  def video2video(ref_img, source_video, size=512, steps=25, length=60, seed=42):
216
+ print('=====Start processing======')
217
+
218
  cfg = 3.5
219
  fi_step = 3
220