added functionality to translate

2023-08-28 15:18:57 +02:00
parent bf8ee9acca
commit 5be187998e
1 changed files with 3 additions and 0 deletions
@@ -166,6 +166,9 @@ class Transcriber:
        _possible_kwargs = Whisper.transcribe.__code__.co_varnames
        
        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