Fix

Softcatala · Sep 16, 2024 · 6f43e50 · 6f43e50
1 parent 96bdbd1
commit 6f43e50
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/open_dubbing/dubbing.py b/open_dubbing/dubbing.py
@@ -126,6 +126,7 @@ def __init__(
         self.utterance_metadata = None
         self._number_of_steps = number_of_steps
         self.tts = tts
+        self.stt = sst
         self.device = device
         self.cpu_threads = cpu_threads
 

diff --git a/open_dubbing/main.py b/open_dubbing/main.py
@@ -209,10 +209,10 @@ def main():
             stt = SpeechToTextWhisperTransfomers(args.device, args.cpu_threads)
         else:
             stt = SpeechToTextFasterWhisper(args.device, args.cpu_threads)
-    elif args == "faster-whisper":
+    elif args.stt == "faster-whisper":
         stt = SpeechToTextFasterWhisper(args.device, args.cpu_threads)
     else:
-        stt = SpeechToTextFasterWhisper(args.device, args.cpu_threads)
+        stt = SpeechToTextWhisperTransfomers(args.device, args.cpu_threads)
 
     check_languages(args.source_language, args.target_language, tts, stt)