h4d35 commited on
Commit
f7413c9
·
1 Parent(s): e355bf4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -1,12 +1,15 @@
1
- from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
2
  import soundfile as sf
3
  import torch
4
  import gradio as gr
5
 
6
 
7
  # load model and processor
8
- processor = Wav2Vec2Processor.from_pretrained("h4d35/Wav2Vec2-hi")
9
- model = Wav2Vec2ForCTC.from_pretrained("h4d35/Wav2Vec2-hi")
 
 
 
10
 
11
  # define function to read in sound file
12
  def map_to_array(file):
 
1
+ #from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
2
  import soundfile as sf
3
  import torch
4
  import gradio as gr
5
 
6
 
7
  # load model and processor
8
+ from transformers import AutoProcessor, AutoModelForCTC
9
+
10
+ processor = AutoProcessor.from_pretrained("h4d35/Wav2Vec2-hi")
11
+
12
+ model = AutoModelForCTC.from_pretrained("h4d35/Wav2Vec2-hi")
13
 
14
  # define function to read in sound file
15
  def map_to_array(file):