bofenghuang commited on
Commit
4bf0c36
1 Parent(s): f4bfa42
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -24,7 +24,7 @@ model-index:
24
  name: Automatic Speech Recognition
25
  type: automatic-speech-recognition
26
  dataset:
27
- name: Common Voice 11
28
  type: mozilla-foundation/common_voice_11_0
29
  args: fr
30
  metrics:
@@ -130,7 +130,7 @@ import torchaudio
130
 
131
  from transformers import AutoModelForCTC, Wav2Vec2ProcessorWithLM
132
 
133
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
134
 
135
  model = AutoModelForCTC.from_pretrained("bhuang/asr-wav2vec2-french").to(device)
136
  processor_with_lm = Wav2Vec2ProcessorWithLM.from_pretrained("bhuang/asr-wav2vec2-french")
@@ -162,7 +162,7 @@ import torchaudio
162
 
163
  from transformers import AutoModelForCTC, Wav2Vec2Processor
164
 
165
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
166
 
167
  model = AutoModelForCTC.from_pretrained("bhuang/asr-wav2vec2-french").to(device)
168
  processor = Wav2Vec2Processor.from_pretrained("bhuang/asr-wav2vec2-french")
 
24
  name: Automatic Speech Recognition
25
  type: automatic-speech-recognition
26
  dataset:
27
+ name: Common Voice 11.0
28
  type: mozilla-foundation/common_voice_11_0
29
  args: fr
30
  metrics:
 
130
 
131
  from transformers import AutoModelForCTC, Wav2Vec2ProcessorWithLM
132
 
133
+ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
134
 
135
  model = AutoModelForCTC.from_pretrained("bhuang/asr-wav2vec2-french").to(device)
136
  processor_with_lm = Wav2Vec2ProcessorWithLM.from_pretrained("bhuang/asr-wav2vec2-french")
 
162
 
163
  from transformers import AutoModelForCTC, Wav2Vec2Processor
164
 
165
+ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
166
 
167
  model = AutoModelForCTC.from_pretrained("bhuang/asr-wav2vec2-french").to(device)
168
  processor = Wav2Vec2Processor.from_pretrained("bhuang/asr-wav2vec2-french")