|
import gradio as gr |
|
from transformers import pipeline |
|
from pydub import AudioSegment |
|
import os |
|
import speech_recognition as sr |
|
|
|
model_name = "voidful/wav2vec2-xlsr-multilingual-56" |
|
model0 = pipeline(task="automatic-speech-recognition", |
|
model=model_name) |
|
|
|
|
|
model_name = "SLPL/Sharif-wav2vec2" |
|
model2 = pipeline(task="automatic-speech-recognition", |
|
model=model_name) |
|
model_name = "ghofrani/common8" |
|
model1 = pipeline(task="automatic-speech-recognition", |
|
model=model_name) |
|
|
|
import json |
|
def predict_fa(speech,model): |
|
if model== "SLPL/Sharif-wav2vec2": |
|
text = model2(speech,return_timestamps="word" ) |
|
elif model== "ghofrani/common8": |
|
text = model1(speech,return_timestamps="word" ) |
|
elif model== "voidful/wav2vec2-xlsr-multilingual-56": |
|
text = model0(speech,return_timestamps="word" ) |
|
return [text['text'],json.dumps(text)] |
|
|
|
|
|
def convert_to_wav(filename): |
|
filenameObj=os.path.splitext(filename) |
|
audio = AudioSegment.from_file(filename,format=filenameObj[1].replace(".","")) |
|
new_filename = filenameObj[0] + ".wav" |
|
while os.path.exists(new_filename): |
|
new_filename = os.path.splitext(new_filename)[0]+"(1)"+ ".wav" |
|
audio.export(new_filename, format="wav") |
|
print(f"Converting {filename} to {new_filename}...") |
|
return new_filename |
|
def g_rec(audio_File ,language): |
|
r = sr.Recognizer() |
|
|
|
if not os.path.splitext(audio_File)[1]==".wav": |
|
audio_File=convert_to_wav(audio_File) |
|
hellow=sr.AudioFile(audio_File) |
|
with hellow as source: |
|
audio = r.record(source) |
|
try: |
|
s = r.recognize_google(audio,language =language) |
|
res= "Text: "+s |
|
except Exception as e: |
|
res= "Exception: "+str(e) |
|
return res |
|
|
|
|
|
|
|
|
|
|
|
with gr.Blocks() as demo: |
|
gr.Markdown("multilingual Speech Recognition") |
|
|
|
with gr.Tab("Persian models"): |
|
inputs_speech_fa =gr.Audio(source="upload", type="filepath", optional=True,label="Upload your audio:") |
|
inputs_model_fa =gr.inputs.Radio(label="Language", choices=["ghofrani/common8","SLPL/Sharif-wav2vec2","voidful/wav2vec2-xlsr-multilingual-56"]) |
|
output_transcribe1_fa = gr.Textbox(label="Transcribed text:") |
|
output_transcribe1_fa1 = gr.Textbox(label="Transcribed text with timestamps:") |
|
transcribe_audio1_fa= gr.Button("Submit") |
|
with gr.Tab("google"): |
|
gr.Markdown("set your speech language") |
|
inputs_speech1 =[ |
|
gr.Audio(source="upload", type="filepath"), |
|
gr.Dropdown(choices=["af-ZA","am-ET","ar-AE","ar-BH","ar-DZ","ar-EG","ar-IL","ar-IQ","ar-JO","ar-KW","ar-LB","ar-MA","ar-MR","ar-OM","ar-PS","ar-QA","ar-SA","ar-TN","ar-YE","az-AZ","bg-BG","bn-BD","bn-IN","bs-BA","ca-ES","cs-CZ","da-DK","de-AT","de-CH","de-DE","el-GR","en-AU","en-CA","en-GB","en-GH","en-HK","en-IE","en-IN","en-KE","en-NG","en-NZ","en-PH","en-PK","en-SG","en-TZ","en-US","en-ZA","es-AR","es-BO","es-CL","es-CO","es-CR","es-DO","es-EC","es-ES","es-GT","es-HN","es-MX","es-NI","es-PA","es-PE","es-PR","es-PY","es-SV","es-US","es-UY","es-VE","et-EE","eu-ES","fa-IR","fi-FI","fil-PH","fr-BE","fr-CA","fr-CH","fr-FR","gl-ES","gu-IN","hi-IN","hr-HR","hu-HU","hy-AM","id-ID","is-IS","it-CH","it-IT","iw-IL","ja-JP","jv-ID","ka-GE","kk-KZ","km-KH","kn-IN","ko-KR","lo-LA","lt-LT","lv-LV","mk-MK","ml-IN","mn-MN","mr-IN","ms-MY","my-MM","ne-NP","nl-BE","nl-NL","no-NO","pa-Guru-IN","pl-PL","pt-BR","pt-PT","ro-RO","ru-RU","si-LK","sk-SK","sl-SI","sq-AL","sr-RS","su-ID","sv-SE","sw-KE","sw-TZ","ta-IN","ta-LK","ta-MY","ta-SG","te-IN","th-TH","tr-TR","uk-UA","ur-IN","ur-PK","uz-UZ","vi-VN","yue-Hant-HK","zh (cmn-Hans-CN)","zh-TW (cmn-Hant-TW)","zu-ZA"] |
|
,value="fa-IR",label="language code") |
|
] |
|
output_transcribe1 = gr.Textbox(label="output") |
|
transcribe_audio1_go= gr.Button("Submit") |
|
|
|
transcribe_audio1_fa.click(fn=predict_fa, |
|
inputs=[inputs_speech_fa ,inputs_model_fa ], |
|
outputs=[output_transcribe1_fa ,output_transcribe1_fa1 ] ) |
|
|
|
transcribe_audio1_go.click(fn=g_rec, |
|
inputs=inputs_speech1 , |
|
outputs=output_transcribe1 ) |
|
|
|
|
|
if __name__ == "__main__": |
|
demo.launch() |
|
|
