semakoc commited on
Commit
ebbbdc2
·
verified ·
1 Parent(s): b626f3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,12 +1,13 @@
1
  import gradio as gr
2
  import torch
3
  import librosa
4
- from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
 
5
 
6
  # Load Wav2Vec2 Model
7
- MODEL_NAME = "facebook/wav2vec2-large-960h"
8
- processor = Wav2Vec2Processor.from_pretrained(MODEL_NAME)
9
- model = Wav2Vec2ForCTC.from_pretrained(MODEL_NAME)
10
 
11
  def transcribe(audio_file):
12
  """
 
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/ASR_Amis"
9
+ processor =AutoProcessor.from_pretrained(MODEL_NAME)
10
+ model = AutoModelForSpeechSeq2Seq.from_pretrained(MODEL_NAME)
11
 
12
  def transcribe(audio_file):
13
  """