Wav2Vec2 / app.py
s12144251's picture
Update app.py
8232c52 verified
import gradio as gr
from transformers import pipeline
# تحميل نموذج Wav2Vec2 من Hugging Face
model_id = "facebook/wav2vec2-base-960h"
asr = pipeline("automatic-speech-recognition", model=model_id)
# دالة لتحويل الصوت إلى نص
def transcribe_audio(audio):
# استخدام نموذج Wav2Vec2 لتحويل الصوت إلى نص
result = asr(audio)
return result['text']
# إنشاء واجهة Gradio
with gr.Blocks() as demo:
gr.Markdown("# Speech Recognition using Wav2Vec2")
gr.Markdown("قم برفع ملف صوتي لتحويله إلى نص باستخدام نموذج Wav2Vec2")
audio_input = gr.Audio(label="أدخل ملف صوتي", type="filepath")
transcribe_btn = gr.Button("حوّل النص")
text_output = gr.Textbox(label="النص الناتج")
# ربط الأزرار والدوال
transcribe_btn.click(fn=transcribe_audio, inputs=audio_input, outputs=text_output)
# تشغيل التطبيق
demo.launch()