Muhammed_Kotb1 commited on
Commit
1cd840c
1 Parent(s): 61f99c5

try omaradel wav2vec

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,9 +4,9 @@ import jiwer
4
  from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC, Wav2Vec2CTCTokenizer
5
 
6
  # Load the Arabic-specific processor and model
7
- model_name = "omarxadel/hubert-large-arabic-egyptian"
8
  tokenizer = Wav2Vec2CTCTokenizer.from_pretrained(model_name)
9
- processor = Wav2Vec2Processor.from_pretrained(model_name)
10
  model = Wav2Vec2ForCTC.from_pretrained(model_name)
11
 
12
  def transcribe(audio_file):
 
4
  from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC, Wav2Vec2CTCTokenizer
5
 
6
  # Load the Arabic-specific processor and model
7
+ model_name = "omarxadel/wav2vec2-large-xlsr-53-arabic-egyptian"
8
  tokenizer = Wav2Vec2CTCTokenizer.from_pretrained(model_name)
9
+ processor = Wav2Vec2Processor.from_pretrained(model_name, tokenizer=tokenizer)
10
  model = Wav2Vec2ForCTC.from_pretrained(model_name)
11
 
12
  def transcribe(audio_file):