inoculatemedia commited on
Commit
308d965
·
verified ·
1 Parent(s): 4b66926

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
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