daihui.zhang commited on
Commit
f49e9c9
·
1 Parent(s): 93d2288

add export data for test

Browse files
Files changed (1) hide show
  1. transcribe/whisper_llm_serve.py +4 -2
transcribe/whisper_llm_serve.py CHANGED
@@ -58,7 +58,7 @@ class WhisperTranscriptionService(ServeClientBase):
58
  self._transcrible_time_cost = 0.
59
  self._translate_time_cost = 0.
60
  if config.TEST:
61
- self._test_task_event = threading.Event()
62
  self._test_queue = queue.Queue()
63
  self._test_thread = self._start_thread(self.test_data_loop)
64
 
@@ -66,7 +66,7 @@ class WhisperTranscriptionService(ServeClientBase):
66
 
67
  def test_data_loop(self):
68
  writer = TestDataWriter()
69
- while not self._test_task_event.is_set():
70
  test_data = self._test_queue.get()
71
  writer.write(test_data) # Save test_data to CSV
72
 
@@ -307,4 +307,6 @@ class WhisperTranscriptionService(ServeClientBase):
307
  """停止所有处理线程并清理资源"""
308
  self._translate_thread_stop.set()
309
  self._frame_processing_thread_stop.set()
 
 
310
  logger.info(f"Stopping transcription service for client: {self.client_uid}")
 
58
  self._transcrible_time_cost = 0.
59
  self._translate_time_cost = 0.
60
  if config.TEST:
61
+ self._test_task_stop = threading.Event()
62
  self._test_queue = queue.Queue()
63
  self._test_thread = self._start_thread(self.test_data_loop)
64
 
 
66
 
67
  def test_data_loop(self):
68
  writer = TestDataWriter()
69
+ while not self._test_task_stop.is_set():
70
  test_data = self._test_queue.get()
71
  writer.write(test_data) # Save test_data to CSV
72
 
 
307
  """停止所有处理线程并清理资源"""
308
  self._translate_thread_stop.set()
309
  self._frame_processing_thread_stop.set()
310
+ if config.TEST:
311
+ self._test_task_stop.set()
312
  logger.info(f"Stopping transcription service for client: {self.client_uid}")