I am trying to use the model ai4bharat/indictrans2-en-indic-dist-200M BUT IT IS NOT RECOGNISING THE LANGUAGE.
Can someone help me to use it as simple english text to text translation.

from transformers import AutoModelForSeq2SeqLM, AutoTokenizer

MODEL_NAME = "ai4bharat/indictrans2-en-indic-dist-200M"
CACHE_DIR = "./model_cache"

--- The Test Case ---

source_text = "This is a test from the command line"
source_lang = "2en"
target_lang = "2hi"

--- THE CORRECT AND FINAL INPUT FORMAT ---

The model requires both source and target tags prefixed to the text.

input_text = f"<{source_lang}> <{target_lang}> {source_text}"

inputs = tokenizer(input_text, return_tensors="pt")
print("--- Tokenization successful. ---")

# NOTE: No forced_bos_token_id is needed because the target is now specified in the input.
print("\n--- Step 4: Generating translation from model ---")
generated_tokens = model.generate(
    **inputs,
    num_return_sequences=1,
    num_beams=5,
    max_length=256
)

It is failing and not giving that it invalid language. I am missing small part since it may be updated recently.