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

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