Spaces:
Runtime error
Runtime error
from transformers import pipeline, AutoFeatureExtractor, AutoTokenizer, Wav2Vec2ForCTC | |
import gradio as gr | |
import time | |
model_id = 'comodoro/wav2vec2-xls-r-300m-cs-250' | |
feature_extractor = AutoFeatureExtractor.from_pretrained(model_id) | |
model = Wav2Vec2ForCTC.from_pretrained(model_id) | |
tokenizer = AutoTokenizer.from_pretrained(model_id) | |
p = pipeline("automatic-speech-recognition", chunk_length_s=5, model=model, | |
tokenizer=tokenizer, feature_extractor=feature_extractor) | |
def transcribe(audio, state=""): | |
time.sleep(2) | |
text = p(audio)["text"] | |
state += text + " " | |
return state, state | |
gr.Interface( | |
fn=transcribe, | |
inputs=[ | |
gr.inputs.Audio(source="microphone", type="filepath"), | |
"state" | |
], | |
outputs=[ | |
"textbox", | |
"state" | |
], | |
live=True).launch(debug=True) | |