diff --git a/predict.py b/predict.py
index d615236..332df40 100644
--- a/predict.py
+++ b/predict.py
@@ -15,6 +15,7 @@
 from cog import BasePredictor, Path, Input
 
 from demo import load_checkpoints
+from demo import
 from demo import make_animation
 from ffhq_dataset.face_alignment import image_align
 from ffhq_dataset.landmarks_detector import LandmarksDetector