Make Faster Whisper and OpenAI Whisper work

2024-07-03 18:43:27 +02:00
parent b0adecef7a
commit 2d7630a757
7 changed files with 261 additions and 51 deletions
--- a/stt_voice_messages/models/model_settings.py
+++ b/stt_voice_messages/models/model_settings.py
@@ -19,5 +19,10 @@ from dataclasses import dataclass, field

@dataclass
 class OpenAIWhisperSettings:
+    model_size: str = field(default='tiny', init=True)
+    translate_to_english: bool = field(default=False, init=True)
+
+@dataclass
+class FasterWhisperSettings:
    model_size: str = field(default='tiny', init=True)
    translate_to_english: bool = field(default=False, init=True)