inoculatemedia commited on
Commit
90814e1
·
verified ·
1 Parent(s): 0ca2ed5

Update inference_video_enhance.py

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