From 4b93d33eb201bc5ca923f84ce664432b4e6930b2 Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Tue, 9 Feb 2021 09:32:04 +0800
Subject: [PATCH] Release v2 model

---
 inference_img.py   | 2 +-
 inference_video.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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()