👍[Update] Model : CTranslate2のダウンロード処理を追加

This commit is contained in:
misyaguziya
2024-01-17 23:22:00 +09:00
parent 5fb9f5a6ec
commit c20f27fadf
7 changed files with 45 additions and 27 deletions

View File

@@ -24,6 +24,7 @@ from models.transcription.transcription_transcriber import AudioTranscriber
from models.xsoverlay.notification import xsoverlayForVRCT
from models.translation.translation_languages import translation_lang
from models.transcription.transcription_languages import transcription_lang
from models.translation.utils import checkCTranslate2Weight
from config import config
class threadFnc(Thread):
@@ -63,10 +64,15 @@ class Model:
self.speaker_audio_recorder = None
self.speaker_energy_recorder = None
self.speaker_energy_plot_progressbar = None
self.translator = Translator(config.PATH_LOCAL, config.WEIGHT_TYPE)
self.translator = Translator()
if config.USE_TRANSLATION_FEATURE is True:
self.translator.changeCTranslate2Model(config.PATH_LOCAL, config.WEIGHT_TYPE)
self.keyword_processor = KeywordProcessor()
def updateTranslator(self):
def checkCTranslatorCTranslate2ModelWeight(self):
return checkCTranslate2Weight(config.PATH_LOCAL, config.WEIGHT_TYPE)
def changeTranslatorCTranslate2Model(self):
self.translator.changeCTranslate2Model(config.PATH_LOCAL, config.WEIGHT_TYPE)
def resetKeywordProcessor(self):