Update demo/infer.py

demo/infer.py

@@ -76,14 +76,18 @@ class LiveCCDemoInfer:
         """
         # 1. preparation: video_reader, and last processing info
         video_timestamp, last_timestamp = state.get('video_timestamp', 0), state.get('last_timestamp', -1 / self.fps)
-        video_path = state['video_path']
+        video_path = state.get('video_path', None)
+        if not video_path:
+            return
         if video_path not in self._cached_video_readers_with_hw:
             self._cached_video_readers_with_hw[video_path] = get_smart_resized_video_reader(video_path, max_pixels)
             video_reader = self._cached_video_readers_with_hw[video_path][0]
             video_reader.get_frame_timestamp(0)
             state['video_pts'] = torch.from_numpy(video_reader._frame_pts[:, 1])
             state['last_video_pts_index'] = -1
-        video_pts = state['video_pts']
+        video_pts = state.get('video_pts', None)
+        if not video_pts:
+            return
         video_timestamp = min(video_timestamp, video_pts[-1])
         if last_timestamp + self.frame_time_interval > video_pts[-1]:
             state['video_end'] = True