hueso/gajim-plugins
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve multi modul support and refactor

Browse Source
This commit is contained in:
mesonium
2024-07-02 20:05:45 +02:00
committed by hueso
parent aff69e5b40
commit b0adecef7a
6 changed files with 111 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
stt_voice_messages/models/model_settings.py
View File

@@ -19,5 +19,5 @@ from dataclasses import dataclass, field
@dataclass
class OpenAIWhisperSettings:
whisperai_model_size: str = field(default='tiny', init=True)
model_size: str = field(default='tiny', init=True)
translate_to_english: bool = field(default=False, init=True)

6
stt_voice_messages/models/model_template.py
View File

@@ -16,8 +16,6 @@
from abc import ABC, abstractmethod
from pathlib import Path
from gajim.gtk.const import Setting
from ..helper import Results
@@ -26,7 +24,3 @@ class Model(ABC):
@abstractmethod
def transcribe(self, result: Results, audio_file: Path) -> str:
return ''
@abstractmethod
def on_setting(self, setting: Setting):
pass

11
stt_voice_messages/models/openai_whisper.py
View File

@@ -46,11 +46,10 @@ class WhisperModel(Model):
return self._result
def transcribe(self, result: Results, audio_file: Path) -> str:
model = whisper.load_model(self._config['whisperai_model_size'])
log.debug('model size is used = %s', self._config['whisperai_model_size'])
result.text = model.transcribe(audio_file)['text']
model = whisper.load_model(self._config.model_size)
log.debug('model size is used = %s', self._config.model_size)
result.text = model.transcribe(audio_file)['text'] # pyright: ignore [reportAttributeAccessIssue]
def on_setting(self, key, value):
log.debug('key = %s, value = %s', key, value)
self._config[key] = value
def set_config(self, config: OpenAIWhisperSettings) -> None:
self._config = config