create new

app.py

@@ -3,9 +3,12 @@ import os
 import datetime
 import inference
 
+import spaces
+
 example1 = ["sample_data/ref1.jpg", "sample_data/ano.mp3"]
 example2 = ["sample_data/ref2.jpg", "sample_data/rakugo.mp3"]
 
+@spaces.GPU
 def fix_face_video(input_image, input_audio):
 
     dt = datetime.datetime.now() + datetime.timedelta(hours=9)

inference.py

@@ -19,8 +19,6 @@ from pipelines import VExpressPipeline
 from pipelines.utils import draw_kps_image, save_video
 from pipelines.utils import retarget_kps
 
-import spaces
-
 # 引数用ダミークラス
 class args_dum:
 
@@ -164,7 +162,6 @@ def get_scheduler():
     scheduler = DDIMScheduler(**scheduler_kwargs)
     return scheduler
 
-@spaces.GPU
 def fix_face(image, audio, out_path):
     # args = parse_args()
     args = args_dum()

requirements.txt

@@ -1,3 +1,5 @@
+accelerate
+gradio==4.32.1
 torch==2.0.1
 xformers==0.0.22
 diffusers==0.24.0
@@ -12,4 +14,4 @@ transformers==4.30.2
 einops==0.4.1
 tqdm==4.66.1
 av==11.0.0
-opencv-python
+opencv-python