Update inference_video_enhance.py
Browse files
inference_video_enhance.py
CHANGED
@@ -109,7 +109,7 @@ else:
|
|
109 |
videogen.sort(key= lambda x:int(x[:-4]))
|
110 |
lastframe = cv2.imread(os.path.join(args.img, videogen[0]), cv2.IMREAD_UNCHANGED)[:, :, ::-1].copy()
|
111 |
videogen = videogen[1:]
|
112 |
-
|
113 |
h, w, _ = lastframe.shape
|
114 |
|
115 |
vid_out_name = None
|
|
|
109 |
videogen.sort(key= lambda x:int(x[:-4]))
|
110 |
lastframe = cv2.imread(os.path.join(args.img, videogen[0]), cv2.IMREAD_UNCHANGED)[:, :, ::-1].copy()
|
111 |
videogen = videogen[1:]
|
112 |
+
@spaces.GPU(duration=5)
|
113 |
h, w, _ = lastframe.shape
|
114 |
|
115 |
vid_out_name = None
|