From 8534d17ab00de729867bb482ab204854127ed2a1 Mon Sep 17 00:00:00 2001 From: yum Date: Tue, 25 Apr 2023 14:27:23 -0700 Subject: Restore string matching, remove affinity mask Affinity mask no longer affects performance. String matching is still needed for temporal stability in fast-paced long-form transcription tasks. --- Scripts/transcribe.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'Scripts') diff --git a/Scripts/transcribe.py b/Scripts/transcribe.py index 4d3169f..043ca0c 100644 --- a/Scripts/transcribe.py +++ b/Scripts/transcribe.py @@ -15,6 +15,7 @@ import langcodes import pyaudio import numpy as np import steamvr +import string_matcher import sys import threading import time @@ -249,8 +250,9 @@ def transcribeAudio(audio_state, model, use_cpu: bool): last_transcribe_time = time.time() continue - old_text = copy.copy(audio_state.text) - audio_state.text = text + old_text = audio_state.text + audio_state.text = string_matcher.matchStrings(audio_state.text, + text, window_size = 25); now = time.time() print("Transcription ({} seconds): {}".format( -- cgit v1.2.3