Commit
•
7639b35
1
Parent(s):
7f1584d
update
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ from pathlib import Path
|
|
| 7 |
import numpy as np
|
| 8 |
import cv2
|
| 9 |
import torch
|
| 10 |
-
import spaces
|
| 11 |
|
| 12 |
from diffusers import AutoencoderKL, DDIMScheduler
|
| 13 |
from einops import repeat
|
|
@@ -105,7 +105,7 @@ vis = FaceMeshVisualizer()
|
|
| 105 |
|
| 106 |
frame_inter_model = init_frame_interpolation_model()
|
| 107 |
|
| 108 |
-
|
| 109 |
def audio2video(input_audio, ref_img, headpose_video=None, size=512, steps=25, length=150, seed=42):
|
| 110 |
fps = 30
|
| 111 |
cfg = 3.5
|
|
@@ -211,7 +211,7 @@ def audio2video(input_audio, ref_img, headpose_video=None, size=512, steps=25, l
|
|
| 211 |
|
| 212 |
return save_path.replace('_noaudio.mp4', '.mp4'), ref_image_pil
|
| 213 |
|
| 214 |
-
|
| 215 |
def video2video(ref_img, source_video, size=512, steps=25, length=150, seed=42):
|
| 216 |
cfg = 3.5
|
| 217 |
|
|
|
|
| 7 |
import numpy as np
|
| 8 |
import cv2
|
| 9 |
import torch
|
| 10 |
+
#import spaces
|
| 11 |
|
| 12 |
from diffusers import AutoencoderKL, DDIMScheduler
|
| 13 |
from einops import repeat
|
|
|
|
| 105 |
|
| 106 |
frame_inter_model = init_frame_interpolation_model()
|
| 107 |
|
| 108 |
+
#@spaces.GPU(duration=200)
|
| 109 |
def audio2video(input_audio, ref_img, headpose_video=None, size=512, steps=25, length=150, seed=42):
|
| 110 |
fps = 30
|
| 111 |
cfg = 3.5
|
|
|
|
| 211 |
|
| 212 |
return save_path.replace('_noaudio.mp4', '.mp4'), ref_image_pil
|
| 213 |
|
| 214 |
+
#@spaces.GPU(duration=200)
|
| 215 |
def video2video(ref_img, source_video, size=512, steps=25, length=150, seed=42):
|
| 216 |
cfg = 3.5
|
| 217 |
|