Update app.py

app.py

@@ -35,13 +35,8 @@ def do_interpolation(frame1, frame2):
             input_frames, times_to_interpolate, interpolator))
     
     #print(frames)
-    named_frames = []
-    for idx, i in frames:
-        i = Image.fromarray(np.uint8(i))
-        i.save(f'{frame1}_{frame2}_{idx}.jpg')
-        named_frame.append(f'{frame1}_{frame2}_{idx}.jpg')
-    print(named_frames)
-    return frames
+    mediapy.write_video(f"{frame1}_to_{frame2}_out.mp4", frames, fps=12)
+    return f"{frame1}_to_{frame2}_out.mp4"
     
 def get_frames(video_in):
     frames = []
@@ -118,7 +113,11 @@ def infer(video_in):
         if idx < len(frames_list) - 1:
             next_frame = frames_list[idx+1]
             interpolated_frames = do_interpolation(frame, next_frame) # should return a list of 3 interpolated frames
-            result_frames.extend(interpolated_frames)
+            break_interpolated_video = get_frames(interpolated_frames)[0]
+            for j, img in break_interpolated_video:
+                img = Image.open(img)
+                img = Image.save(f"{frame}_to_{next_frame}_{j}")
+                result_frames.append(img)
             print("frames " + str(idx) + " & " + str(idx+1) + "/" + str(n_frame) + ": done;")
     
     final_vid = create_video(result_frames, fps, "interpolated")