Update README.md

README.md

@@ -352,7 +352,7 @@ def inference(batch):
     input_features = processor(audio, sampling_rate=batch["audio"][0]["sampling_rate"], return_tensors="pt").input_features
     input_features = input_features.to(device, dtype=torch_dtype)
     # 2. Auto-regressively generate the predicted token ids
-    pred_ids = model.generate(input_features, max_new_tokens=128)
+    pred_ids = model.generate(input_features, language="ja", max_new_tokens=128)
     # 3. Decode the token ids to the final transcription
     batch["transcription"] = processor.batch_decode(pred_ids, skip_special_tokens=True)
     batch["reference"] = batch[text_column]