fixed bug that language cant be choosen

2023-09-18 15:56:45 +02:00
parent 5385e266cc
commit 8019ce43b0
1 changed files with 3 additions and 0 deletions
@@ -173,6 +173,9 @@ class Transcriber:
        if (task := kwargs.get("task")):
            whisper_kwargs["task"] = task
            
+        if (language := kwargs.get("language")):
+            whisper_kwargs["language"] = language 
+        
        return whisper_kwargs
    
    def __repr__(self) -> str: