Update app.py

app.py

@@ -38,7 +38,7 @@ def do_interpolation(frame1, frame2, interpolation):
     mediapy.write_video(f"{frame1}_to_{frame2}_out.mp4", frames, fps=25)
     return f"{frame1}_to_{frame2}_out.mp4"
     
-def get_frames(video_in, step, name, resize_n):
+def get_frames(video_in, step, name, resize_w):
     frames = []
     cap = cv2.VideoCapture(video_in)
     cframes = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
@@ -70,7 +70,8 @@ def get_frames(video_in, step, name, resize_n):
         ret, frame = cap.read()
         if ret == False:
             break
-        frame = cv2.resize(frame, (int(resize_n), int(resize_n/2.0))) 
+        resize_h = resize_w / 2.0
+        frame = cv2.resize(frame, (int(resize_w), int(resize_h))) 
         cv2.imwrite(f"{name}_{step}{str(i)}.jpg",frame)
         frames.append(f"{name}_{step}{str(i)}.jpg")
         i+=1
@@ -90,11 +91,11 @@ def create_video(frames, fps, type):
     return type + "_result.mp4"
 
     
-def infer(url_in,interpolation,fps_output,resize_n):
+def infer(url_in,interpolation,fps_output,resize_w):
     
     fps_output = logscale(fps_output)
     # 1. break video into frames and get FPS
-    break_vid = get_frames(url_in, "vid_input_frame", "origin", resize_n)
+    break_vid = get_frames(url_in, "vid_input_frame", "origin", resize_w)
     frames_list= break_vid[0]
     fps = break_vid[1]
     print(f"ORIGIN FPS: {fps}")