Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-21 23:57:58 +01:00
Code Issues Packages Projects Releases Wiki Activity

cleanup

Browse Source
This commit is contained in:
EliasVincent 2023-03-09 20:33:00 +01:00
parent 4c72e43bcf
commit 7a03d0bda3
1 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
extensions/whisper_stt/script.py
View File

@ -1,6 +1,6 @@
import gradio as gr
import speech_recognition as sr
import modules.shared as shared
input_hijack = {
'state': False,
@ -16,25 +16,21 @@ def do_stt():
transcription = ""
r = sr.Recognizer()
with sr.Microphone() as source:
print("Say something!")
r.adjust_for_ambient_noise(source)
audio = r.listen(source)
# recognize speech using whisper
try:
transcription = r.recognize_whisper(audio, language="english", model="tiny.en")
print("Whisper thinks you said " + transcription)
except sr.UnknownValueError:
print("Whisper could not understand audio")
except sr.RequestError as e:
print("Could not request results from Whisper")
print("Could not request results from Whisper", e)
# input_modifier(transcription)
input_hijack.update({"state": True, "value": [transcription, transcription]})
return transcription
def ui():
speech_button = gr.Button(value="STT")
output_transcription = gr.Textbox(label="Speech Preview")
speech_button = gr.Button(value="🎙️")
output_transcription = gr.Textbox(label="STT-Preview", placeholder="Speech Preview. Click \"Generate\" to send")
speech_button.click(do_stt, outputs=[output_transcription])