Ivydata
/

whisper-small-japanese

Automatic Speech Recognition

Model card Files Files and versions Community

Kosuke-Szk commited on May 19, 2023

Commit

120634c

·

1 Parent(s): 6b04853

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -25,7 +25,7 @@ MODEL_ID = "Ivydata/whisper-small-japanese"
 SAMPLES = 10
 test_dataset = load_dataset("common_voice", LANG_ID, split=f"test[:{SAMPLES}]")
-processor = WhisperProcessor.from_pretrained("openai/whisper-base")
 model = WhisperForConditionalGeneration.from_pretrained(MODEL_ID)
 model.config.forced_decoder_ids = processor.get_decoder_prompt_ids(
     language="ja", task="transcribe"

 SAMPLES = 10
 test_dataset = load_dataset("common_voice", LANG_ID, split=f"test[:{SAMPLES}]")
+processor = WhisperProcessor.from_pretrained(MODEL_ID)
 model = WhisperForConditionalGeneration.from_pretrained(MODEL_ID)
 model.config.forced_decoder_ids = processor.get_decoder_prompt_ids(
     language="ja", task="transcribe"