Update app.py

app.py

@@ -1,3 +1,31 @@
 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()
 
-gr.load("models/distil-whisper/distil-large-v2").launch()