Update app.py

app.py

@@ -8,6 +8,8 @@ import matplotlib.pyplot as plt
 import matplotlib.animation as animation
 from skimage.transform import resize
 import warnings
+from demo import make_animation
+from skimage import img_as_ubyte
 from demo import load_checkpoints
 
 warnings.filterwarnings("ignore")
@@ -46,4 +48,9 @@ st.write("Hello, World!")
 # display Markdown
 st.write("# Hello, World!")
 
-generator, kp_detector = load_checkpoints(config_path='config/vox-256.yaml',checkpoint_path='vox-cpk.pth.tar')
+generator, kp_detector = load_checkpoints(config_path='config/vox-256.yaml',checkpoint_path='vox-cpk.pth.tar')
+
+predictions = make_animation(source_image, driving_video, generator, kp_detector, relative=True)
+
+#save resulting video
+imageio.mimsave('generated.mp4', [img_as_ubyte(frame) for frame in predictions])