clu-ling
/

whisper-large-v2-japanese-5k-steps

Automatic Speech Recognition

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

elsayedissa commited on Feb 5, 2023

Commit

d55380a

•

1 Parent(s): 1d992ab

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -65,8 +65,8 @@ from transformers import WhisperProcessor, WhisperForConditionalGeneration
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # load the model
-processor = WhisperProcessor.from_pretrained("clu-ling/whisper-large-v2-arabic-5k-steps")
-model = WhisperForConditionalGeneration.from_pretrained("clu-ling/whisper-large-v2-arabic-5k-steps").to(device)
 forced_decoder_ids = processor.get_decoder_prompt_ids(language="ja", task="transcribe")
 # load the dataset

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # load the model
+processor = WhisperProcessor.from_pretrained("clu-ling/whisper-large-v2-japanese-5k-steps")
+model = WhisperForConditionalGeneration.from_pretrained("clu-ling/whisper-large-v2-japanese-5k-steps").to(device)
 forced_decoder_ids = processor.get_decoder_prompt_ids(language="ja", task="transcribe")
 # load the dataset