Update app.py
Browse files
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 |
-
|
|
|
|
|
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)
|