chenjoya commited on
Commit
be0c9b1
·
verified ·
1 Parent(s): 92542aa

Update demo/infer.py

Browse files
Files changed (1) hide show
  1. demo/infer.py +1 -1
demo/infer.py CHANGED
@@ -86,7 +86,7 @@ class LiveCCDemoInfer:
86
  state['video_pts'] = torch.from_numpy(video_reader._frame_pts[:, 1])
87
  state['last_video_pts_index'] = -1
88
  video_pts = state.get('video_pts', None)
89
- if not video_pts:
90
  return
91
  video_timestamp = min(video_timestamp, video_pts[-1])
92
  if last_timestamp + self.frame_time_interval > video_pts[-1]:
 
86
  state['video_pts'] = torch.from_numpy(video_reader._frame_pts[:, 1])
87
  state['last_video_pts_index'] = -1
88
  video_pts = state.get('video_pts', None)
89
+ if video_pts is None:
90
  return
91
  video_timestamp = min(video_timestamp, video_pts[-1])
92
  if last_timestamp + self.frame_time_interval > video_pts[-1]: