Mohssinibra commited on
Commit
856e48b
·
verified ·
1 Parent(s): 7df5ee9
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ processor = Wav2Vec2Processor.from_pretrained("boumehdi/wav2vec2-large-xlsr-moro
9
 
10
  # Load translation model
11
  translation_model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
12
- translation_tokenizer = MBart50Tokenizer.from_pretrained("facebook/mbart-large-50-many-to-many-mmt", src_lang="ar")
13
 
14
  def transcribe_audio(audio):
15
  # Load the audio file from Gradio interface
 
9
 
10
  # Load translation model
11
  translation_model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
12
+ translation_tokenizer = MBart50Tokenizer.from_pretrained("facebook/mbart-large-50-many-to-many-mmt", src_lang="ar_AR")
13
 
14
  def transcribe_audio(audio):
15
  # Load the audio file from Gradio interface