yavuzkomecoglu commited on
Commit
f2a5fa2
1 Parent(s): f34d3d2

updated model

Browse files
Files changed (1) hide show
  1. utils.py +3 -3
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 = Wav2Vec2Processor.from_pretrained("patrickvonplaten/wav2vec2-large-...-common_voice-tr-demo")
32
- self.model = Wav2Vec2ForCTC.from_pretrained("mpatrickvonplaten/wav2vec2-large-...-common_voice-tr-demo").to(self.device)
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