LuoYiSULIXAY's picture
frist commit
bc3643e verified
raw
history blame
949 Bytes
import gradio as gr
from transformers import pipeline, WhisperProcessor, WhisperForConditionalGeneration
# load mode
model_name = "openai/whisper-medium"
processor = WhisperProcessor.from_pretrained(model_name,language="lo")
model = WhisperForConditionalGeneration.from_pretrained(model_dir)
asr = pipeline(
"automatic-speech-recognition",
model= "LuoYiSULIXAY/whisper-lao-finetuned_laonlp",
tokenizer=processor.tokenizer,
feature_extractor=processor.feature_extractor,
device=-1
)
def transcribe(audio):
result = asr(audio, generate_kwargs={"language": "lao", "task": "transcribe"})
return result["text"]
demo = gr.Interface(
fn=transcribe,
inputs=gr.Audio(type="filepath",streaming=False), # ✅ 正确写法
outputs="text",
title="Whisper Lao",
description="Realtime demo for Lao speech recognition using a fine-tuned Whisper model.",
)
demo.launch(share=True)