Update app.py

app.py

@@ -74,8 +74,8 @@ def get_frames(video_in, step, name, resize_w):
           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")
+        cv2.imwrite(f"{name}_{step}{str(i)}.png",frame)
+        frames.append(f"{name}_{step}{str(i)}.png")
         i+=1
     
     cap.release()
@@ -88,7 +88,7 @@ def get_frames(video_in, step, name, resize_w):
 def create_video(frames, fps, type):
     print("building video result")
     clip = ImageSequenceClip(frames, fps=fps)
-    clip.write_videofile(type + "_result.mp4", fps=fps, codec='libx265', ffmpeg_params=['-c:v', 'libx265'], audio = False)
+    clip.write_videofile(type + "_result.mp4", fps=fps, codec='libx265', ffmpeg_params=['-c:v', 'libx265', '-x265-params', 'crf=28' ], audio = False)
     
     return type + "_result.mp4"
 
@@ -120,8 +120,8 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
             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")
+                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}")