mohamed1ai commited on
Commit
06be46e
1 Parent(s): 34948bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
  from transformers import pipeline, WhisperProcessor, WhisperForConditionalGeneration
3
 
4
- processor = WhisperProcessor.from_pretrained("facebook/whisper-large")
5
- model = WhisperForConditionalGeneration.from_pretrained("facebook/whisper-large")
6
 
7
  def transcrire_audio(audio, prompt):
8
  input_features = processor(audio, return_tensors="pt").input_features
 
1
  import gradio as gr
2
  from transformers import pipeline, WhisperProcessor, WhisperForConditionalGeneration
3
 
4
+ processor = WhisperProcessor.from_pretrained("distil-whisper/distil-large-v2")
5
+ model = WhisperForConditionalGeneration.from_pretrained("distil-whisper/distil-large-v2")
6
 
7
  def transcrire_audio(audio, prompt):
8
  input_features = processor(audio, return_tensors="pt").input_features