From 5be187998e5bd9fd7e0c2ef944d25c9b1aa83a3d Mon Sep 17 00:00:00 2001
From: Jaikinator <schmieder.jacob@web.de>
Date: Mon, 28 Aug 2023 15:18:57 +0200
Subject: [PATCH] added functionality to translate

---
 autotranscript/transcriber.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/autotranscript/transcriber.py b/autotranscript/transcriber.py
index 81787da..e319372 100644
--- a/autotranscript/transcriber.py
+++ b/autotranscript/transcriber.py
@@ -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