fix bugs

libs/film/predict.py

@@ -7,12 +7,12 @@ import mediapy
 from PIL import Image
 import cog
 
-from eval import interpolator, util
+from .eval import interpolator, util
 
 _UINT8_MAX_F = float(np.iinfo(np.uint8).max)
 
 
-class Predictor(cog.Predictor):
+class Predictor(cog.BasePredictor):
     def setup(self):
         import tensorflow as tf
         print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))
@@ -21,26 +21,7 @@ class Predictor(cog.Predictor):
         # Batched time.
         self.batch_dt = np.full(shape=(1,), fill_value=0.5, dtype=np.float32)
 
-    @cog.input(
-        "frame1",
-        type=Path,
-        help="The first input frame",
-    )
-    @cog.input(
-        "frame2",
-        type=Path,
-        help="The second input frame",
-    )
-    @cog.input(
-        "times_to_interpolate",
-        type=int,
-        default=1,
-        min=1,
-        max=8,
-        help="Controls the number of times the frame interpolator is invoked If set to 1, the output will be the "
-             "sub-frame at t=0.5; when set to > 1, the output will be the interpolation video with "
-             "(2^times_to_interpolate + 1) frames, fps of 30.",
-    )
+    
     def predict(self, frame1, frame2, times_to_interpolate):
         INPUT_EXT = ['.png', '.jpg', '.jpeg']
         assert os.path.splitext(str(frame1))[-1] in INPUT_EXT and os.path.splitext(str(frame2))[-1] in INPUT_EXT, \