daihui.zhang
commited on
Commit
·
f49e9c9
1
Parent(s):
93d2288
add export data for test
Browse files
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.
|
| 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.
|
| 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}")
|