update to nso

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
 
-model_name = "dsfsi/ss-en-m2m100-gov"
+model_name = "dsfsi/nso-en-m2m100-gov"
 tokenizer = M2M100Tokenizer.from_pretrained(model_name)
 model = M2M100ForConditionalGeneration.from_pretrained(model_name)
 
-tokenizer.src_lang = "ss"
+tokenizer.src_lang = "nso"
 
 model.config.forced_bos_token_id = tokenizer.get_lang_id("en")
 
@@ -20,24 +20,20 @@ def translate(inp):
 description = """
 <p>
 <center>
-Siswati to English Translation
+Northern Sotho to English Translation
 </center>
 </p>
 """
-article = "<p style='text-align: center'><a href='https://huggingface.co/dsfsi/ss-en-m2m100-gov' target='_blank'>by dsfsi</a></p>"
+article = "<p style='text-align: center'><a href='https://huggingface.co/dsfsi/nso-en-m2m100-gov' target='_blank'>by dsfsi</a></p>"
 
-examples = [
-    ["Ngiyabonga kakhulu ngesipho osinike sona."],
-    ["Ukuthula kuhlale kuyindlela ephilayo yempilo yethu."]
-]
 
 iface = gr.Interface(
     fn=translate,
-    title="🌸Siswati to English Translation🌸",
+    title="Northern Sotho to English Translation",
     description=description,
     article=article,
     examples=examples,
-    inputs=gr.components.Textbox(lines=5, placeholder="Enter Siswati text (maximum 5 lines)", label="Input"),
+    inputs=gr.components.Textbox(lines=5, placeholder="Enter Sotho text (maximum 5 lines)", label="Input"),
     outputs="text"
 )