abhishek HF staff commited on
Commit
331ab4c
1 Parent(s): a3257ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -107,7 +107,7 @@ def inference(video, image):
107
  kp_detector,
108
  relative=True,
109
  adapt_movement_scale=True,
110
- cpu=False,
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=False,
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(