Spaces:

abhishek
/

first-order-motion-model

Build error

abhishek HF staff commited on Dec 21, 2021

Commit

c714bcb

•

1 Parent(s): eeaf709

use gpu

Files changed (3) hide show

app.py CHANGED Viewed

@@ -102,7 +102,7 @@ def inference(video, image):
         kp_detector,
         relative=True,
         adapt_movement_scale=True,
-        cpu=True,
     )
     imageio.mimsave("result.mp4", [img_as_ubyte(frame) for frame in predictions], fps=fps)
     imageio.mimsave("driving.mp4", [img_as_ubyte(frame) for frame in driving_video], fps=fps)
@@ -120,7 +120,7 @@ examples = [["bella_porch.mp4", "julien.png"]]
 generator, kp_detector = load_checkpoints(
     config_path="config/vox-256.yaml",
     checkpoint_path="weights/vox-adv-cpk.pth.tar",
-    cpu=True,
 )
 iface = gr.Interface(

         kp_detector,
         relative=True,
         adapt_movement_scale=True,
+        cpu=False,
     )
     imageio.mimsave("result.mp4", [img_as_ubyte(frame) for frame in predictions], fps=fps)
     imageio.mimsave("driving.mp4", [img_as_ubyte(frame) for frame in driving_video], fps=fps)
 generator, kp_detector = load_checkpoints(
     config_path="config/vox-256.yaml",
     checkpoint_path="weights/vox-adv-cpk.pth.tar",
+    cpu=False,
 )
 iface = gr.Interface(

final.mp4 ADDED Viewed

+version https://git-lfs.github.com/spec/v1
+oid sha256:61787c7b967f5c0ab7ec755c440d2af029acb51ebce6ae3692c55666e29de1aa
+size 680850

gradio_queue.db CHANGED Viewed

Binary files a/gradio_queue.db and b/gradio_queue.db differ