Spaces:
Runtime error
Runtime error
| import librosa | |
| import torch | |
| from transformers import Wav2Vec2ForCTC, Wav2Vec2Tokenizer | |
| import gradio as gr | |
| from transformers import pipeline | |
| import IPython.display as display | |
| import soundfile as sf | |
| def speech_text(audio_file): | |
| tokenizer = Wav2Vec2Tokenizer.from_pretrained("facebook/wav2vec2-base-960h") | |
| model = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h") | |
| speech, rate = librosa.load(audio_file,sr=16000) | |
| display.Audio(audio_file, autoplay=True) | |
| print(rate) | |
| input_values = tokenizer(speech, return_tensors ='pt').input_values | |
| #Store logits (non-normalized predictions) | |
| logits = model(input_values).logits | |
| #Store predicted id's | |
| predicted_ids = torch.argmax(logits, dim =-1) | |
| transcriptions = tokenizer.decode(predicted_ids[0]) | |
| return transcriptions | |
| iface = gr.Interface(speech_text,inputs="audio",outputs="text",title='Sakil Transcription',description="Transcription") | |
| iface.launch(inline=False) | |