From 06533729ed09e5abe40d4217825c4c2ad8da8746 Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Sun, 15 Nov 2020 16:48:09 +0800
Subject: [PATCH] Fix last frame

---
 inference_mp4_2x.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/inference_mp4_2x.py b/inference_mp4_2x.py
index 9df0dbd..43c290e 100644
--- a/inference_mp4_2x.py
+++ b/inference_mp4_2x.py
@@ -53,7 +53,7 @@ while success:
             output.write(lastframe)
             output.write(mid[:h, :w])
     if args.montage:
-        output.write(np.concatenate((frame, frame), 1))
+        output.write(np.concatenate((lastframe, lastframe), 1))
     else:
-        output.write(frame)
+        output.write(lastframe)
 output.release()