Update demo/infer.py

demo/infer.py

@@ -54,7 +54,7 @@ class LiveCCDemoInfer:
     @torch.inference_mode()
     def live_cc(
         self,
-        query: str,
+        message: str,
         state: dict,
         max_pixels: int = 384 * 28 * 28,
         default_query: str = 'Please describe the video.',
@@ -129,12 +129,12 @@ class LiveCCDemoInfer:
                     {"type": "video", "video": clip}
                 ]
             }
-            if not query and not state.get('query', None):
-                query = default_query
+            if not message and not state.get('message', None):
+                message = default_query
                 logger.warning(f'No query provided, use default_query={default_query}')
-            if query and state.get('query', None) != query:
-                message['content'].append({"type": "text", "text": query})
-                state['query'] = query
+            if message and state.get('message', None) != message:
+                message['content'].append({"type": "text", "text": message})
+                state['message'] = message
             texts = self.processor.apply_chat_template([message], tokenize=False, add_generation_prompt=True, return_tensors='pt')
             past_ids = state.get('past_ids', None)
             if past_ids is not None: