Spaces:
Paused
Paused
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]:
|