Spaces:
Running
on
Zero
Running
on
Zero
Update demo/infer.py
Browse files- 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
|
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]:
|