validated whispercpp working, fixed args.listen not being saved due to brainworms

This commit is contained in:
mrq 2023-03-03 07:23:10 +00:00
parent 740b5587df
commit 9fb4aa7917

View File

@ -898,8 +898,11 @@ def prepare_dataset( files, outdir, language=None, progress=None ):
torchaudio.save(f"{outdir}/{sliced_name}", sliced_waveform, sampling_rate) torchaudio.save(f"{outdir}/{sliced_name}", sliced_waveform, sampling_rate)
transcription.append(f"{sliced_name}|{segment['text'].strip()}")
idx = idx + 1 idx = idx + 1
line = f"{sliced_name}|{segment['text'].strip()}"
transcription.append(line)
with open(f'{outdir}/train.txt', 'a', encoding="utf-8") as f:
f.write(f'{line}\n')
with open(f'{outdir}/whisper.json', 'w', encoding="utf-8") as f: with open(f'{outdir}/whisper.json', 'w', encoding="utf-8") as f:
f.write(json.dumps(results, indent='\t')) f.write(json.dumps(results, indent='\t'))
@ -1350,7 +1353,7 @@ def update_args( listen, share, check_for_updates, models_from_local_only, low_v
def save_args_settings(): def save_args_settings():
global args global args
settings = { settings = {
'listen': None if args.listen else args.listen, 'listen': None if not args.listen else args.listen,
'share': args.share, 'share': args.share,
'low-vram':args.low_vram, 'low-vram':args.low_vram,
'check-for-updates':args.check_for_updates, 'check-for-updates':args.check_for_updates,
@ -1587,7 +1590,7 @@ def load_whisper_model(name=None, progress=None, language=b'en'):
notify_progress(f"Loading Whisper model: {args.whisper_model}", progress) notify_progress(f"Loading Whisper model: {args.whisper_model}", progress)
if args.whisper_cpp: if args.whisper_cpp:
from whispercpp import Whisper from whispercpp import Whisper
whisper_model = Whisper(name, models_dir='./models/', language=language) whisper_model = Whisper(name, models_dir='./models/', language=language.encode('ascii'))
else: else:
import whisper import whisper
whisper_model = whisper.load_model(args.whisper_model) whisper_model = whisper.load_model(args.whisper_model)