testing

tim-roethig-db · May 18, 2024 · 1ff67a9 · 1ff67a9
1 parent 43e6064
commit 1ff67a9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/amondin/main.py b/amondin/main.py
@@ -25,7 +25,7 @@ def transcribe(
     :return:
     """
 
-    print(f"Running on {device}.")
+    print(f"Running on {device}...")
 
     print("Diarizing speakers...")
     diarized_speakers = diarize_speakers(

diff --git a/amondin/speech2text.py b/amondin/speech2text.py
@@ -6,7 +6,7 @@
 from transformers import WhisperProcessor, WhisperForConditionalGeneration
 
 
-def speech2text(audio: dict, device: str, model: str = "openai/whisper-tiny", language: str = "german") -> str:
+def speech2text(audio: dict, device: str, model: str, language: str) -> str:
     """
     Translate audio to text
     :param device: Device to run the model on [cpu, cuda or cuda:x]