Update app.py

app.py

@@ -8,6 +8,7 @@ import matplotlib.pyplot as plt
 import matplotlib.animation as animation
 from skimage.transform import resize
 import warnings
+from demo import load_checkpoints
 
 warnings.filterwarnings("ignore")
 source_image = imageio.imread('harold.jpg')
@@ -44,4 +45,6 @@ st.write("Hello, World!")
 st.write(display(source_image, driving_video).to_html5_video(), unsafe_allow_html=True)
 
 # display Markdown
-st.write("# Hello, World!")
+st.write("# Hello, World!")
+
+generator, kp_detector = load_checkpoints(config_path='config/vox-256.yaml',checkpoint_path='vox-cpk.pth.tar')