Spaces:
Runtime error
Runtime error
Commit
·
f2a5fa2
1
Parent(s):
f34d3d2
updated model
Browse files
utils.py
CHANGED
@@ -2,7 +2,7 @@
|
|
2 |
import librosa
|
3 |
import torch
|
4 |
import torchaudio
|
5 |
-
from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
|
6 |
from datasets import load_dataset
|
7 |
|
8 |
import numpy as np
|
@@ -28,8 +28,8 @@ class SpeechRecognition:
|
|
28 |
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
29 |
#self.processor = Wav2Vec2Processor.from_pretrained("m3hrdadfi/wav2vec2-large-xlsr-turkish")
|
30 |
#self.model = Wav2Vec2ForCTC.from_pretrained("m3hrdadfi/wav2vec2-large-xlsr-turkish").to(self.device)
|
31 |
-
self.processor =
|
32 |
-
self.model = Wav2Vec2ForCTC.from_pretrained("
|
33 |
|
34 |
return self
|
35 |
|
|
|
2 |
import librosa
|
3 |
import torch
|
4 |
import torchaudio
|
5 |
+
from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor, AutoTokenizer
|
6 |
from datasets import load_dataset
|
7 |
|
8 |
import numpy as np
|
|
|
28 |
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
29 |
#self.processor = Wav2Vec2Processor.from_pretrained("m3hrdadfi/wav2vec2-large-xlsr-turkish")
|
30 |
#self.model = Wav2Vec2ForCTC.from_pretrained("m3hrdadfi/wav2vec2-large-xlsr-turkish").to(self.device)
|
31 |
+
self.processor = AutoTokenizer.from_pretrained("patrickvonplaten/wav2vec2-large-...-common_voice-tr-demo")
|
32 |
+
self.model = Wav2Vec2ForCTC.from_pretrained("patrickvonplaten/wav2vec2-large-...-common_voice-tr-demo").to(self.device)
|
33 |
|
34 |
return self
|
35 |
|