PKUWilliamYang commited on
Commit
5427b01
1 Parent(s): 25a8215

Update vtoonify_model.py

Browse files
Files changed (1) hide show
  1. vtoonify_model.py +1 -1
vtoonify_model.py CHANGED
@@ -161,7 +161,7 @@ class Model():
161
  video_cap = cv2.VideoCapture(video)
162
  if video_cap.get(7) == 0:
163
  video_cap.release()
164
- return 'default.mp4' instyle, 'Error: fail to load the video.'
165
  num = min(300, int(video_cap.get(7)))
166
  if self.device == 'cpu':
167
  num = min(100, num)
 
161
  video_cap = cv2.VideoCapture(video)
162
  if video_cap.get(7) == 0:
163
  video_cap.release()
164
+ return 'default.mp4', instyle, 'Error: fail to load the video.'
165
  num = min(300, int(video_cap.get(7)))
166
  if self.device == 'cpu':
167
  num = min(100, num)