version1

app.py

@@ -7,7 +7,6 @@ import numpy as np
 import matplotlib.pyplot as plt
 import matplotlib.animation as animation
 from skimage.transform import resize
-from IPython.display import HTML
 import warnings
 import os
 from model import load_checkpoints
@@ -26,8 +25,8 @@ source_image_path = './assets/source.png'
 driving_video_path = './assets/driving.mp4'
 output_video_path = './generated.mp4'
 config_path = './config/vox-256.yaml'
-checkpoint_path = 'checkpoints/vox.pth.tar'
-checkpoint_path = 'vox.pth.tar'
+checkpoint_path = 'models/vox.pth.tar'
+
 predict_mode = 'relative' # ['standard', 'relative', 'avd']
 find_best_frame = False # when use the relative mode to animate a face, use 'find_best_frame=True' can get better quality result