Update app.py

app.py

@@ -59,7 +59,7 @@ def get_frames(video_in, step, name):
         clip_resized = clip.resize(height=576)
         clip_resized.write_videofile("video_resized.mp4", fps=clip.fps, bitrate="12000k")
     
-    print("video resized to 512 height")
+    print("video resized to 576 height")
     
     # Opens the Video file with CV2
     cap= cv2.VideoCapture("video_resized.mp4")
@@ -71,8 +71,9 @@ def get_frames(video_in, step, name):
         ret, frame = cap.read()
         if ret == False:
             break
-        cv2.imwrite(f"{name}_{step}{str(i)}.jpg",frame)
-        frames.append(f"{name}_{step}{str(i)}.jpg")
+        # use use png to avoid any compression artifact
+        cv2.imwrite(f"{name}_{step}{str(i)}.png",frame)
+        frames.append(f"{name}_{step}{str(i)}.png")
         i+=1
     
     cap.release()
@@ -125,8 +126,9 @@ def infer(secret_token, video_in_base64, interpolation, fps_output):
             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}")
-                os.rename(img, f"{frame}_to_{next_frame}_{j}.jpg")
-                result_frames.append(f"{frame}_to_{next_frame}_{j}.jpg")
+                # we use png to avoid any compression artifact
+                os.rename(img, f"{frame}_to_{next_frame}_{j}.png")
+                result_frames.append(f"{frame}_to_{next_frame}_{j}.png")
             
             print("frames " + str(idx) + " & " + str(idx+1) + "/" + str(n_frame) + ": done;")
             #print(f"CURRENT FRAMES: {result_frames}")