Update app.py

app.py

@@ -38,7 +38,7 @@ def do_interpolation(frame1, frame2):
     mediapy.write_video(f"{frame1}_to_{frame2}_out.mp4", frames, fps=12)
     return f"{frame1}_to_{frame2}_out.mp4"
     
-def get_frames(video_in):
+def get_frames(video_in, step):
     frames = []
     #resize the video
     clip = VideoFileClip(video_in)
@@ -65,8 +65,8 @@ def get_frames(video_in):
         ret, frame = cap.read()
         if ret == False:
             break
-        cv2.imwrite('kang'+str(i)+'.jpg',frame)
-        frames.append('kang'+str(i)+'.jpg')
+        cv2.imwrite(f"{step}{str(i)}.jpg",frame)
+        frames.append(f"{step}{str(i)}.jpg")
         i+=1
     
     cap.release()
@@ -92,7 +92,7 @@ def infer(video_in):
     
     
     # 1. break video into frames and get FPS
-    break_vid = get_frames(video_in)
+    break_vid = get_frames(video_in, vid_input_frame)
     frames_list= break_vid[0]
     fps = break_vid[1]
     #n_frame = int(trim_value*fps)
@@ -113,7 +113,7 @@ 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
-            break_interpolated_video = get_frames(interpolated_frames)
+            break_interpolated_video = get_frames(interpolated_frames, interpol)
             print(break_interpolated_video[0])
             for j, img in enumerate(break_interpolated_video[0]):
                 print(f"IMG:{img}")