Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -107,7 +107,7 @@ def inference(video, image):
|
|
107 |
kp_detector,
|
108 |
relative=True,
|
109 |
adapt_movement_scale=True,
|
110 |
-
cpu=
|
111 |
)
|
112 |
imageio.mimsave("result.mp4", [img_as_ubyte(frame) for frame in predictions], fps=fps)
|
113 |
imageio.mimsave("driving.mp4", [img_as_ubyte(frame) for frame in driving_video], fps=fps)
|
@@ -125,7 +125,7 @@ examples = [["bella_porch.mp4", "julien.png"]]
|
|
125 |
generator, kp_detector = load_checkpoints(
|
126 |
config_path="config/vox-256.yaml",
|
127 |
checkpoint_path="weights/vox-adv-cpk.pth.tar",
|
128 |
-
cpu=
|
129 |
)
|
130 |
|
131 |
iface = gr.Interface(
|
|
|
107 |
kp_detector,
|
108 |
relative=True,
|
109 |
adapt_movement_scale=True,
|
110 |
+
cpu=True,
|
111 |
)
|
112 |
imageio.mimsave("result.mp4", [img_as_ubyte(frame) for frame in predictions], fps=fps)
|
113 |
imageio.mimsave("driving.mp4", [img_as_ubyte(frame) for frame in driving_video], fps=fps)
|
|
|
125 |
generator, kp_detector = load_checkpoints(
|
126 |
config_path="config/vox-256.yaml",
|
127 |
checkpoint_path="weights/vox-adv-cpk.pth.tar",
|
128 |
+
cpu=True,
|
129 |
)
|
130 |
|
131 |
iface = gr.Interface(
|