PKUWilliamYang commited on
Commit
4dd3399
1 Parent(s): b954edf

Update vtoonify_model.py

Browse files
Files changed (1) hide show
  1. vtoonify_model.py +1 -1
vtoonify_model.py CHANGED
@@ -219,7 +219,7 @@ class Model():
219
  if aligned_video is None:
220
  return 'default.mp4', 'Opps, something wrong with the input. Please go to Step 2 and Rescale Video again.'
221
  video_cap = cv2.VideoCapture(aligned_video)
222
- if instyle is None or aligned_face is None or video_cap.get(7) == 0:
223
  video_cap.release()
224
  return 'default.mp4', 'Opps, something wrong with the input. Please go to Step 2 and Rescale Video again.'
225
  if exstyle is None:
 
219
  if aligned_video is None:
220
  return 'default.mp4', 'Opps, something wrong with the input. Please go to Step 2 and Rescale Video again.'
221
  video_cap = cv2.VideoCapture(aligned_video)
222
+ if instyle is None or aligned_video is None or video_cap.get(7) == 0:
223
  video_cap.release()
224
  return 'default.mp4', 'Opps, something wrong with the input. Please go to Step 2 and Rescale Video again.'
225
  if exstyle is None: