clr commited on
Commit
738c923
·
1 Parent(s): c565886

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import soundfile as sf
3
  from scipy import signal
4
  import numpy as np
5
  import torch, torchaudio
6
- from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
7
 
8
  MODEL_IS="carlosdanielhernandezmena/wav2vec2-large-xlsr-53-icelandic-ep10-1000h"
9
  MODEL_FO="carlosdanielhernandezmena/wav2vec2-large-xlsr-53-faroese-100h"
@@ -17,6 +17,9 @@ model_fo = Wav2Vec2ForCTC.from_pretrained(MODEL_FO).to(device)
17
  processor_fo = Wav2Vec2Processor.from_pretrained(MODEL_FO)
18
 
19
 
 
 
 
20
  def readwav(a_f):
21
  wav, sr = sf.read(a_f, dtype=np.float32)
22
  if len(wav.shape) == 2:
@@ -38,7 +41,9 @@ def recc(audio_file,model,processor):
38
 
39
 
40
  def recis(audio_file):
41
- return recc(audio_file,model_is,processor_is)
 
 
42
 
43
  def recfo(audio_file):
44
  return recc(audio_file,model_fo,processor_fo)
 
3
  from scipy import signal
4
  import numpy as np
5
  import torch, torchaudio
6
+ from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor, pipeline
7
 
8
  MODEL_IS="carlosdanielhernandezmena/wav2vec2-large-xlsr-53-icelandic-ep10-1000h"
9
  MODEL_FO="carlosdanielhernandezmena/wav2vec2-large-xlsr-53-faroese-100h"
 
17
  processor_fo = Wav2Vec2Processor.from_pretrained(MODEL_FO)
18
 
19
 
20
+ pipe_is = pipeline(model=MODEL_IS)
21
+
22
+
23
  def readwav(a_f):
24
  wav, sr = sf.read(a_f, dtype=np.float32)
25
  if len(wav.shape) == 2:
 
41
 
42
 
43
  def recis(audio_file):
44
+ output = pipe(audio_file, chunk_length_s=10, stride_length_s=(4, 2))
45
+ return output
46
+ #return recc(audio_file,model_is,processor_is)
47
 
48
  def recfo(audio_file):
49
  return recc(audio_file,model_fo,processor_fo)