Update README.md

README.md

@@ -8,6 +8,11 @@ tags:
 
 # Cascaded Japanese Speech2Text Translation
 This is a pipeline for speech-to-text translation from Japanese speech to any target language text based on the cascaded approach, that consists of ASR and translation.
+The pipeline employs [kotoba-tech/kotoba-whisper-v2.0](https://huggingface.co/kotoba-tech/kotoba-whisper-v2.0) for ASR (Japanese speech -> Japanese text)
+and [facebook/nllb-200-3.3B](https://huggingface.co/facebook/nllb-200-3.3B) for text translation.
+The input must be Japanese speech, while the translation can be in any languages NLLB trained on. Please find the all available languages and their language codes
+[here](https://github.com/facebookresearch/flores/blob/main/flores200/README.md#languages-in-flores-200).
+
 
 ## Usage
 Here is an example to translate Japanese speech into English text translation.
@@ -24,7 +29,7 @@ from transformers import pipeline
 pipe = pipeline(
     model="japanese-asr/ja-cascaded-s2t-translation",
     model_kwargs={"attn_implementation": "sdpa"},
-    model_translation="facebook/nllb-200-distilled-600M",
+    model_translation="facebook/nllb-200-3.3B",
     tgt_lang="eng_Latn",
     chunk_length_s=15,
     trust_remote_code=True,
@@ -33,3 +38,12 @@ pipe = pipeline(
 # translate
 output = pipe("./sample_ja.flac")
 ```
+
+
+Other NLLB models can be used by setting `model_translation` such as following.
+- [facebook/nllb-200-3.3B](https://huggingface.co/facebook/nllb-200-3.3B)
+- [facebook/nllb-200-distilled-600M](https://huggingface.co/facebook/nllb-200-distilled-600M)
+- [facebook/nllb-200-distilled-1.3B](https://huggingface.co/facebook/nllb-200-distilled-1.3B)
+- [facebook/nllb-200-1.3B](https://huggingface.co/facebook/nllb-200-1.3B)
+
+