SillyTavern/SillyTavern-extras
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern-Extras.git synced 2026-05-01 03:41:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #96 from Tony-sama/neo

Browse Source 
Restored speech recognition streaming mode as an independant module.
This commit is contained in:
Cohee
2023-07-31 20:42:40 +03:00
committed by GitHub
parent 403c107dc7 912c4335c2
commit d47cb539ad
3 changed files with 135 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
server.py
View File

@@ -326,7 +326,7 @@ Compress(app) # compress responses
app.config["MAX_CONTENT_LENGTH"] = 100 * 1024 * 1024
if "vosk-stt" in modules:
print("Initializing Vosk STT streaming")
print("Initializing Vosk speech-recognition (from ST request file)")
vosk_model_path = (
args.stt_vosk_model_path
if args.stt_vosk_model_path
@@ -338,7 +338,7 @@ if "vosk-stt" in modules:
app.add_url_rule("/api/speech-recognition/vosk/process-audio", view_func=vosk_module.process_audio, methods=["POST"])
if "whisper-stt" in modules:
print("Initializing Whisper STT streaming")
print("Initializing Whisper speech-recognition (from ST request file)")
whisper_model_path = (
args.stt_whisper_model_path
if args.stt_whisper_model_path
@@ -349,6 +349,18 @@ if "whisper-stt" in modules:
whisper_module.model = whisper_module.load_model(file_path=whisper_model_path)
app.add_url_rule("/api/speech-recognition/whisper/process-audio", view_func=whisper_module.process_audio, methods=["POST"])
if "streaming-stt" in modules:
print("Initializing vosk/whisper speech-recognition (from extras server microphone)")
whisper_model_path = (
args.stt_whisper_model_path
if args.stt_whisper_model_path
else None)
import modules.speech_recognition.streaming_module as streaming_module
streaming_module.whisper_model, streaming_module.vosk_model = streaming_module.load_model(file_path=whisper_model_path)
app.add_url_rule("/api/speech-recognition/streaming/record-and-transcript", view_func=streaming_module.record_and_transcript, methods=["POST"])
def require_module(name):
def wrapper(fn):
@wraps(fn)