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 |
|