Update app.py
Browse files
app.py
CHANGED
@@ -58,9 +58,7 @@ def upscale_video(video_path, scale_factor, progress=gr.Progress()):
|
|
58 |
pil_frame = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
|
59 |
|
60 |
inputs = ImageLoader.load_image(pil_frame)
|
61 |
-
preds = model(inputs)
|
62 |
-
output_frame = ImageLoader.save_image(preds, mode='RGB').convert("RGB")
|
63 |
-
|
64 |
output_frame = ImageLoader.to_pil(preds)
|
65 |
video_writer.write(cv2.cvtColor(np.array(output_frame), cv2.COLOR_RGB2BGR))
|
66 |
|
|
|
58 |
pil_frame = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
|
59 |
|
60 |
inputs = ImageLoader.load_image(pil_frame)
|
61 |
+
preds = model(inputs)
|
|
|
|
|
62 |
output_frame = ImageLoader.to_pil(preds)
|
63 |
video_writer.write(cv2.cvtColor(np.array(output_frame), cv2.COLOR_RGB2BGR))
|
64 |
|