Hugging Face's logo

Spaces:
chronopt-research
/
SwinTExCo
Sleeping

App Files Community
1
duongttr commited on
Commit
19d9b4e
·
1 Parent(s): 7a00143

Update change color space

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -27,6 +27,7 @@ def video_colorization(video_path, ref_image, progress=gr.Progress()):
27
  # Init progress bar
28
 
29
  for colorized_frame, _ in zip(model.predict_video(video_reader, ref_image), progress.tqdm(range(num_frames), desc="Colorizing video", unit="frames")):
 
30
  video_writer.write(colorized_frame)
31
 
32
  # for i in progress.tqdm(range(1000)):
 
27
  # Init progress bar
28
 
29
  for colorized_frame, _ in zip(model.predict_video(video_reader, ref_image), progress.tqdm(range(num_frames), desc="Colorizing video", unit="frames")):
30
+ colorized_frame = cv2.cvtColor(colorized_frame, cv2.COLOR_RGB2BGR)
31
  video_writer.write(colorized_frame)
32
 
33
  # for i in progress.tqdm(range(1000)):