ovos-audio-transformer-plugin-fasterwhisper

OpenVoiceOS · May 4, 2023 · 70de39c · 70de39c
1 parent c4c53ae
commit 70de39c
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -15,6 +15,8 @@ available models are `"tiny.en", "tiny", "base.en", "base", "small.en", "small",
 
 eg, to use Large model with GPU
 
+To use Whisper as STT
+
 ```json
   "stt": {
     "module": "ovos-stt-plugin-fasterwhisper",
@@ -28,6 +30,20 @@ eg, to use Large model with GPU
 
 ```
 
+To use Whisper for lang detection  (ovos-dinkum-listener only)
+
+
+```json
+  "listener": {
+    "audio_transformers": {
+        "ovos-audio-transformer-plugin-fasterwhisper": {
+            "model": "small"
+        }
+    }
+  }
+
+```
+
 ## Models
 
 Models will be auto downloaded by faster whisper on plugin load

diff --git a/ovos_stt_plugin_fasterwhisper/__init__.py b/ovos_stt_plugin_fasterwhisper/__init__.py
@@ -10,7 +10,7 @@
 class FasterWhisperLangClassifier(AudioTransformer):
     def __init__(self, config=None):
         config = config or {}
-        super().__init__("fasterwhisper_lang", 10, config)
+        super().__init__("ovos-audio-transformer-plugin-fasterwhisper", 10, config)
         model = self.config.get("model")
         if not model:
             model = "small.en"

diff --git a/setup.py b/setup.py
@@ -51,7 +51,7 @@ def required(requirements_file):
 
 PLUGIN_ENTRY_POINT = 'ovos-stt-plugin-fasterwhisper = ovos_stt_plugin_fasterwhisper:FasterWhisperSTT'
 CONFIG_ENTRY_POINT = 'ovos-stt-plugin-fasterwhisper.config = ovos_stt_plugin_fasterwhisper:FasterWhisperSTTConfig'
-LANG_PLUGIN_ENTRY_POINT = 'ovos-stt-plugin-fasterwhisper = ovos_stt_plugin_fasterwhisper:FasterWhisperLangClassifier'
+LANG_PLUGIN_ENTRY_POINT = 'ovos-audio-transformer-plugin-fasterwhisper = ovos_stt_plugin_fasterwhisper:FasterWhisperLangClassifier'
 
 
 setup(