add m2m model

app.py

@@ -35,6 +35,9 @@ model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M"
 mbart_m2en_model = EasyNMT("mbart50_m2en")
 mbart_en2m_model = EasyNMT("mbart50_en2m")
 
+# Initialize m2m_100 models
+m2m_model = EasyNMT("m2m_100_1.2B")
+
 
 class myTheme(Base):
     def __init__(
@@ -169,6 +172,17 @@ def mbart_trans(article, target_language):
             return mbart_m2en_model.translate(article, target_lang="en")
         else:
             return mbart_en2m_model.translate(article, target_lang="zh")
+    else:
+        return "Error: You chose the same language as the article detected language. Please reselect language and try again."
+
+
+def m2m_trans(article, target_language):
+    result_lang = detect_lang(article)
+
+    if result_lang != target_language:
+        return m2m_model.translate(article)
+    else:
+        return "Error: You chose the same language as the article detected language. Please reselect language and try again."
 
 
 def translate(article, toolkit, target_language):
@@ -178,6 +192,8 @@ def translate(article, toolkit, target_language):
         translated = nllb_trans(article, target_language)
     elif toolkit == "MBART":
         translated = mbart_trans(article, target_language)
+    elif toolkit == "M2M":
+        translated = m2m_trans(article, target_language)
 
     return translated
 
@@ -187,7 +203,7 @@ myTheme = myTheme()
 with gr.Blocks(theme=myTheme) as demo:
     article = gr.Textbox(label="Article")
     toolkit_select = gr.Radio(
-        ["OPUS", "NLLB", "MBART"], label="Select Translation Model", value="OPUS"
+        ["OPUS", "NLLB", "MBART", "M2M"], label="Select Translation Model", value="OPUS"
     )
     lang_select = gr.Radio(["English", "Chinese"], label="Select Desired Language")
     result = gr.Textbox(label="Translated Result")