use gpu

app.py

@@ -102,7 +102,7 @@ def inference(video, image):
         kp_detector,
         relative=True,
         adapt_movement_scale=True,
-        cpu=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)
@@ -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,
+    cpu=False,
 )
 
 iface = gr.Interface(

final.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:61787c7b967f5c0ab7ec755c440d2af029acb51ebce6ae3692c55666e29de1aa
+size 680850