diff --git a/inference_img.py b/inference_img.py
index feef968..8087aca 100644
--- a/inference_img.py
+++ b/inference_img.py
@@ -3,7 +3,7 @@ import cv2
 import torch
 import argparse
 from torch.nn import functional as F
-from model.RIFE_HD import Model
+from model.RIFE_HDv2 import Model
 import warnings
 warnings.filterwarnings("ignore")
 
diff --git a/inference_video.py b/inference_video.py
index 517dc66..7f175d9 100644
--- a/inference_video.py
+++ b/inference_video.py
@@ -76,7 +76,7 @@ assert (not args.video is None or not args.img is None)
 if not args.img is None:
     args.png = True
 
-from model.RIFE_HD import Model
+from model.RIFE_HDv2 import Model
 model = Model()
 model.load_model(os.path.join(os.path.dirname(os.path.realpath(__file__)), 'train_log'), -1)
 model.eval()