bofenghuang
/

asr-wav2vec2-ctc-french

Automatic Speech Recognition

hf-asr-leaderboard

robust-speech-event

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

bofenghuang commited on Dec 18, 2022

Commit

4bf0c36

•

1 Parent(s): f4bfa42

up

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -24,7 +24,7 @@ model-index:
       name: Automatic Speech Recognition
       type: automatic-speech-recognition
     dataset:
-      name: Common Voice 11
       type: mozilla-foundation/common_voice_11_0
       args: fr
     metrics:
@@ -130,7 +130,7 @@ import torchaudio
 from transformers import AutoModelForCTC, Wav2Vec2ProcessorWithLM
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = AutoModelForCTC.from_pretrained("bhuang/asr-wav2vec2-french").to(device)
 processor_with_lm = Wav2Vec2ProcessorWithLM.from_pretrained("bhuang/asr-wav2vec2-french")
@@ -162,7 +162,7 @@ import torchaudio
 from transformers import AutoModelForCTC, Wav2Vec2Processor
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = AutoModelForCTC.from_pretrained("bhuang/asr-wav2vec2-french").to(device)
 processor = Wav2Vec2Processor.from_pretrained("bhuang/asr-wav2vec2-french")

       name: Automatic Speech Recognition
       type: automatic-speech-recognition
     dataset:
+      name: Common Voice 11.0
       type: mozilla-foundation/common_voice_11_0
       args: fr
     metrics:
 from transformers import AutoModelForCTC, Wav2Vec2ProcessorWithLM
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 model = AutoModelForCTC.from_pretrained("bhuang/asr-wav2vec2-french").to(device)
 processor_with_lm = Wav2Vec2ProcessorWithLM.from_pretrained("bhuang/asr-wav2vec2-french")
 from transformers import AutoModelForCTC, Wav2Vec2Processor
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 model = AutoModelForCTC.from_pretrained("bhuang/asr-wav2vec2-french").to(device)
 processor = Wav2Vec2Processor.from_pretrained("bhuang/asr-wav2vec2-french")