semakoc commited on
Commit
11e5741
·
verified ·
1 Parent(s): e60a557

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,13 +1,13 @@
1
  import gradio as gr
2
  import torch
3
  import librosa
4
- from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
5
 
6
 
7
  # Load Wav2Vec2 Model
8
  MODEL_NAME = "eleferrand/xlsr53_Amis"
9
- processor =AutoProcessor.from_pretrained(MODEL_NAME)
10
- model = AutoModelForSpeechSeq2Seq.from_pretrained(MODEL_NAME)
11
 
12
  def transcribe(audio_file):
13
  """
 
1
  import gradio as gr
2
  import torch
3
  import librosa
4
+ from transformers import Wave2VecProcessor, AutoModelForCTC
5
 
6
 
7
  # Load Wav2Vec2 Model
8
  MODEL_NAME = "eleferrand/xlsr53_Amis"
9
+ processor =Wave2VecProcessor.from_pretrained(MODEL_NAME)
10
+ model = AutoModelForCTC.from_pretrained(MODEL_NAME)
11
 
12
  def transcribe(audio_file):
13
  """