Update app.py

app.py

@@ -70,8 +70,10 @@ def get_frames(video_in, step, name, resize_w):
         ret, frame = cap.read()
         if ret == False:
             break
-        resize_h = resize_w / 2.0
-        frame = cv2.resize(frame, (int(resize_w), int(resize_h))) 
+        if resize_w > 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
@@ -114,7 +116,7 @@ def infer(url_in,interpolation,fps_output,resize_n):
         if idx < len(frames_list) - 1:
             next_frame = frames_list[idx+1]
             interpolated_frames = do_interpolation(frame, next_frame, interpolation) # should return a list of 3 interpolated frames
-            break_interpolated_video = get_frames(interpolated_frames, "interpol",f"{idx}_")
+            break_interpolated_video = get_frames(interpolated_frames, "interpol", f"{idx}_", 0)
             print(break_interpolated_video[0])
             for j, img in enumerate(break_interpolated_video[0][0:len(break_interpolated_video[0])-1]):
                 print(f"IMG:{img}")