Feat translate option

coruscant187 · Feb 8, 2023 · bfc2177 · bfc2177
1 parent dce3669
commit bfc2177
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/app/config.py b/app/config.py
@@ -29,6 +29,7 @@
     "compression_ratio_threshold": 2.4,
     "condition_on_previous_text": True,
     "verbose": False,
+    "task": False
 }
 WHISPER_SETTINGS_FILE = DATA_DIR / ".whisper_settings.json"
 

diff --git a/app/core.py b/app/core.py
@@ -49,6 +49,11 @@ def _transcribe(self, audio_path: str, whisper_model: str, **whisper_args):
         else:
             whisper_args["temperature"] = [whisper_args["temperature"]]
 
+        if whisper_args["task"]:
+            whisper_args["task"] = "translate"
+        else:
+            whisper_args["task"] = "transcribe"
+
         del whisper_args["temperature_increment_on_fallback"]
 
         transcript = transcriber.transcribe(