change token for src lang

app.py

@@ -1,18 +1,25 @@
 import gradio as gr
-from transformers import pipeline, M2M100Tokenizer
+from transformers import M2M100Tokenizer, M2M100ForConditionalGeneration, pipeline
 
 model_name = "dsfsi/nr-en-m2m100-gov"
 tokenizer = M2M100Tokenizer.from_pretrained(model_name)
+model = M2M100ForConditionalGeneration.from_pretrained(model_name)
 
-print(tokenizer.lang_code_to_token)
+tokenizer.src_lang = "nr"
 
-translater_nr_en = pipeline("translation", model=model_name, src_lang="nr", tgt_lang="en")
+model.config.forced_bos_token_id = tokenizer.get_lang_id("en")
+
+translater_nr_en = pipeline(
+    "translation",
+    model=model,
+    tokenizer=tokenizer,
+)
 
 def translate(inp):
-    # Translate from isiNdebele to English
     res = translater_nr_en(inp, max_length=512, early_stopping=True)[0]['translation_text']
     return res
 
+# Update description and article as needed
 description = """
 <p>
 <center>
@@ -38,3 +45,4 @@ iface = gr.Interface(
 )
 
 iface.launch(enable_queue=True)
+