freealise commited on
Commit
104e59c
·
verified ·
1 Parent(s): 59fa0b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -88,9 +88,9 @@ def create_video(frames, fps, type):
88
  print("building video result")
89
  imgs = []
90
  for j, img in enumerate(frames):
91
- imgs.append(cv2.cvtColor(cv2.imread(img).astype(np.uint8), cv2.COLOR_BGR2RGBA))
92
 
93
- mediapy.write_video(type + "_result.mp4", imgs, fps=fps, codec='hevc', encoded_format='yuva444p')
94
  return type + "_result.mp4"
95
 
96
 
 
88
  print("building video result")
89
  imgs = []
90
  for j, img in enumerate(frames):
91
+ imgs.append(cv2.cvtColor(cv2.imread(img).astype(np.uint8), cv2.COLOR_BGR2RGB))
92
 
93
+ mediapy.write_video(type + "_result.mp4", imgs, fps=fps)
94
  return type + "_result.mp4"
95
 
96