Spaces:
Runtime error
Runtime error
File size: 1,032 Bytes
e80aa38 7ad92e4 e80aa38 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import gradio as gr
from transformers import WhisperProcessor, WhisperForConditionalGeneration
processor = WhisperProcessor.from_pretrained("distil-whisper/distil-large-v2")
model = WhisperForConditionalGeneration.from_pretrained("distil-whisper/distil-large-v2")
def transcrire_audio(audio, prompt):
input_features = processor(audio, return_tensors="pt").input_features
output_without_prompt = model.generate(input_features)
transcription_sans_prompt = processor.decode(output_without_prompt[0])
prompt_ids = processor.get_prompt_ids(prompt)
output_with_prompt = model.generate(input_features, prompt_ids=prompt_ids)
transcription_avec_prompt = processor.decode(output_with_prompt[0])
return {
"Transcription sans prompt": transcription_sans_prompt,
"Transcription avec prompt": transcription_avec_prompt
}
iface = gr.Interface(
fn=transcrire_audio,
inputs=["audio", "text"],
outputs=["text", "text"],
live=True,
interpretation="default"
)
iface.launch()
|