patrickvonplaten commited on
Commit
4ce47f3
1 Parent(s): d09cb7d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -103,8 +103,9 @@ wer = load_metric("wer")
103
 
104
  processor = Wav2Vec2Processor.from_pretrained('gchhablani/wav2vec2-large-xlsr-eo')
105
  model = Wav2Vec2ForCTC.from_pretrained('gchhablani/wav2vec2-large-xlsr-eo')
 
106
 
107
- chars_to_ignore_regex = '[\,\?\.\!\-\;\:\"\“\%\‘\”\�\„\«\(\»\)\’\']'
108
  resampler = torchaudio.transforms.Resample(48_000, 16_000)
109
 
110
  # Preprocessing the datasets.
 
103
 
104
  processor = Wav2Vec2Processor.from_pretrained('gchhablani/wav2vec2-large-xlsr-eo')
105
  model = Wav2Vec2ForCTC.from_pretrained('gchhablani/wav2vec2-large-xlsr-eo')
106
+ model.to("cuda")
107
 
108
+ chars_to_ignore_regex = '[\\,\\?\\.\\!\\-\\;\\:\\"\\“\\%\\‘\\”\\�\\„\\«\\(\\»\\)\\’\\']'
109
  resampler = torchaudio.transforms.Resample(48_000, 16_000)
110
 
111
  # Preprocessing the datasets.