Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-03-30 14:28:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge 3d126013d7005bd23fef116fe4466246ed0d5296 into 517a43ecd132a2089d85f4ebc044728a71d49f6e

Browse Source
This commit is contained in:
Bob Lin 2025-01-25 06:00:22 +00:00 committed by GitHub
commit 88fb1e13da
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
whisper/utils.py
View File

@ -3,7 +3,8 @@ import os
import re
import sys
import zlib
from typing import Callable, List, Optional, TextIO
from pathlib import Path
from typing import Callable, List, Optional, TextIO, Union
system_encoding = sys.getdefaultencoding()
@ -89,16 +90,21 @@ class ResultWriter:
self.output_dir = output_dir
def __call__(
self, result: dict, audio_path: str, options: Optional[dict] = None, **kwargs
):
audio_basename = os.path.basename(audio_path)
audio_basename = os.path.splitext(audio_basename)[0]
self,
result: dict,
audio_path: Union[str, Path],
options: Optional[dict] = None,
**kwargs,
) -> str:
if not isinstance(audio_path, Path):
audio_path = Path(audio_path)
output_path = os.path.join(
self.output_dir, audio_basename + "." + self.extension
self.output_dir, audio_path.with_suffix(self.extension).name
)
with open(output_path, "w", encoding="utf-8") as f:
self.write_result(result, file=f, options=options, **kwargs)
return output_path
def write_result(
self, result: dict, file: TextIO, options: Optional[dict] = None, **kwargs