Update inference_img.py

inference_img.py

@@ -5,7 +5,6 @@ import argparse
 from torch.nn import functional as F
 import warnings
 
-
 def main():
         
     warnings.filterwarnings("ignore")
@@ -18,7 +17,7 @@ def main():
     
     parser = argparse.ArgumentParser(description='Interpolation for a pair of images')
     parser.add_argument('--img', dest='img', nargs=2, required=True)
-    parser.add_argument('--exp', default=4, type=int, required=True)
+    parser.add_argument('--exp', default=4, type=int)
     parser.add_argument('--ratio', default=0, type=float, help='inference ratio between two images with 0 - 1 range')
     parser.add_argument('--rthreshold', default=0.02, type=float, help='returns image when actual ratio falls in given range threshold')
     parser.add_argument('--rmaxcycles', default=8, type=int, help='limit max number of bisectional cycles')
@@ -29,10 +28,10 @@ def main():
     try:
         try:
             try:
-                from model.RIFE_HDv2 import Model
+                from train_log.RIFE_HDv3 import Model
                 model = Model()
                 model.load_model(args.modelDir, -1)
-                print("Loaded v2.x HD model.")
+                print("Loaded v3.x HD model.")
             except:
                 from train_log.RIFE_HDv3 import Model
                 model = Model()