Update app.py

app.py

@@ -1,18 +1,18 @@
 import gradio as gr
-from transformers import WhisperProcessor, WhisperForConditionalGeneration
+from transformers import pipeline, WhisperProcessor, WhisperForConditionalGeneration
 
-processor = WhisperProcessor.from_pretrained("distil-whisper/distil-large-v2")
-model = WhisperForConditionalGeneration.from_pretrained("distil-whisper/distil-large-v2")
+processor = WhisperProcessor.from_pretrained("facebook/whisper-large")
+model = WhisperForConditionalGeneration.from_pretrained("facebook/whisper-large")
 
 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], skip_special_tokens=True)
+    transcription_sans_prompt = processor.batch_decode(output_without_prompt, skip_special_tokens=True)[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], skip_special_tokens=True)
+    transcription_avec_prompt = processor.batch_decode(output_with_prompt, skip_special_tokens=True)[0]
 
     return {
         "Transcription sans prompt": transcription_sans_prompt,
@@ -27,4 +27,3 @@ iface = gr.Interface(
 )
 
 iface.launch()
-