acmyu commited on
Commit
3aa6ee8
·
1 Parent(s): 85ddb46

film output frames and thumbnails

Browse files
Files changed (2) hide show
  1. libs/film/predict.py +2 -6
  2. main.py +1 -1
libs/film/predict.py CHANGED
@@ -66,10 +66,6 @@ class Predictor(cog.BasePredictor):
66
  frames = list(
67
  util.interpolate_recursively_from_files(
68
  input_frames, times_to_interpolate, self.interpolator))
69
- print('Interpolated frames generated, saving now as output video.')
70
 
71
- ffmpeg_path = util.get_ffmpeg_path()
72
- mediapy.set_ffmpeg(ffmpeg_path)
73
- out_path = Path(tempfile.mkdtemp()) / "out.mp4"
74
- mediapy.write_video(str(out_path), frames, fps=30)
75
- return out_path
 
66
  frames = list(
67
  util.interpolate_recursively_from_files(
68
  input_frames, times_to_interpolate, self.interpolator))
69
+ print('Interpolated frames generated.')
70
 
71
+ return frames
 
 
 
 
main.py CHANGED
@@ -1230,7 +1230,7 @@ def run_interpolate_frames(frame1, frame2, times_to_interp):
1230
 
1231
  results = film.predict(frame1, frame2, times_to_interp)
1232
  print(results)
1233
- return [], []
1234
 
1235
 
1236
  """
 
1230
 
1231
  results = film.predict(frame1, frame2, times_to_interp)
1232
  print(results)
1233
+ return results, getThumbnails(results)
1234
 
1235
 
1236
  """