cjones/WhisperLive
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add srt_file_path option

Browse Source
This commit is contained in:
makaveli10
2024-02-20 11:53:45 +05:30
parent 2e9f67ba0b
commit dc22b7da9f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
whisper_live/client.py
+2 -2
View File
@@ -284,7 +284,7 @@ class Client:
assert self.last_response_recieved assert self.last_response_recieved
while time.time() - self.last_response_recieved < self.disconnect_if_no_response_for: while time.time() - self.last_response_recieved < self.disconnect_if_no_response_for:
continue continue
self.send_packet_to_server(Client.END_OF_AUDIO.encode('utf-8')) # Ensure it's sent as bytes self.send_packet_to_server(Client.END_OF_AUDIO.encode('utf-8'))
if self.server_backend == "faster_whisper": if self.server_backend == "faster_whisper":
self.write_srt_file(self.srt_file_path) self.write_srt_file(self.srt_file_path)
self.stream.close() self.stream.close()
@@ -507,7 +507,7 @@ class TranscriptionClient:
``` ```
""" """
def __init__(self, host, port, lang=None, translate=False, model="small", use_vad=True): def __init__(self, host, port, lang=None, translate=False, model="small", use_vad=True):
self.client = Client(host, port, lang, translate, model, use_vad=use_vad) self.client = Client(host, port, lang, translate, model, srt_file_path="output.srt", use_vad=use_vad)
def __call__(self, audio=None, hls_url=None): def __call__(self, audio=None, hls_url=None):
""" """