added functionality to translate

This commit is contained in:
Jaikinator
2023-08-28 15:18:57 +02:00
parent bf8ee9acca
commit 5be187998e
+3
View File
@@ -167,6 +167,9 @@ class Transcriber:
whisper_kwargs = {k: v for k, v in kwargs.items() if k in _possible_kwargs} whisper_kwargs = {k: v for k, v in kwargs.items() if k in _possible_kwargs}
if (task := kwargs.get("task")):
whisper_kwargs["task"] = task
return whisper_kwargs return whisper_kwargs
def __repr__(self) -> str: def __repr__(self) -> str: