Update inference_img.py

inference_img.py

@@ -26,29 +26,19 @@ def main():
     args = parser.parse_args()
     
     try:
-        try:
-            try:
-                from train_log.RIFE_HDv3 import Model
-                model = Model()
-                model.load_model(args.modelDir, -1)
-                print("Loaded v3.x HD model.")
-            except:
-                from train_log.RIFE_HDv3 import Model
-                model = Model()
-                model.load_model(args.modelDir, -1)
-                print("Loaded v3.x HD model.")
-        except:
-            from model.RIFE_HD import Model
-            model = Model()
-            model.load_model(args.modelDir, -1)
-            print("Loaded v1.x HD model")
+        from train_log.RIFE_HDv3 import Model
+        model = Model()
+        model.load_model(args.modelDir, -1)
+        print("Loaded RIFE_HDv3 model.")
     except:
-        from model.RIFE import Model
+        from train_log.IFNet_HDv3 import Model
         model = Model()
         model.load_model(args.modelDir, -1)
-        print("Loaded ArXiv-RIFE model")
+        print("Loaded IFNet_HDv3 model.")
+    
     model.eval()
     model.device()
+
     
     if args.img[0].endswith('.exr') and args.img[1].endswith('.exr'):
         img0 = cv2.imread(args.img[0], cv2.IMREAD_COLOR | cv2.IMREAD_ANYDEPTH)