Add model selection dropdown for alignment mappers

app.py

@@ -2,11 +2,12 @@ import gradio as gr
 from gradio_rich_textbox import RichTextbox
 
 from helper.text_preprocess import space_punc
+from helper.alignment_mappers import select_model
 from helper.pos_taggers import select_pos_tagger
 from helper.translators import select_translator
 
 
-def bn_postagger(src, translator, tagger):
+def bn_postagger(src, translator, model_name, tagger):
     """
     Bangla PoS Tagger
     """
@@ -15,7 +16,9 @@ def bn_postagger(src, translator, tagger):
 
     tgt_base, tgt = select_translator(src, translator)
 
-    result, pos_accuracy = select_pos_tagger(src, tgt, tagger)
+    model_name = select_model(model_name)
+
+    result, pos_accuracy = select_pos_tagger(src, tgt, model_name, tagger)
 
     return tgt_base, result, pos_accuracy
     
@@ -61,6 +64,10 @@ with gr.Blocks(css="styles.css") as demo:
                     choices=["Google", "BanglaNMT", "MyMemory"], 
                     label="Select a Translator"
                 ),
+                gr.Dropdown(
+                    choices=["Google-mBERT (Base-Multilingual)", "Neulab-AwesomeAlign (Bn-En-0.5M)"], 
+                    label="Select a Model"
+                ),
                 gr.Dropdown(
                     choices=["spaCy", "NLTK", "Flair", "TextBlob"], 
                     label="Select a PoS Tagger"
@@ -74,19 +81,54 @@ with gr.Blocks(css="styles.css") as demo:
             outputs = [
                 gr.Textbox(label="English Translation"), 
                 RichTextbox(label="PoS Tags"),
-                gr.Textbox(label="Overall PoS Tagging Accuracy")
+                gr.Textbox(label="PoS Tagging Accuracy (Based on Unknown(UNK) Tags)")
             ]
 
     btn.click(bn_postagger, inputs, outputs)
 
     gr.Examples([
-        ["বাংলাদেশ দক্ষিণ এশিয়ার একটি সার্বভৌম রাষ্ট্র।", "Google", "NLTK"],
-        ["বাংলাদেশের সংবিধানিক নাম কি?", "Google", "spaCy"],
-        ["বাংলাদেশের সাংবিধানিক নাম গণপ্রজাতন্ত্রী বাংলাদেশ।", "Google", "TextBlob"],
-        ["তিনজনের কেউই বাবার পথ ধরে প্রযুক্তি দুনিয়ায় হাঁটেননি।", "Google", "spaCy"],
-        ["তিনজনের কেউই বাবার পথ ধরে প্রযুক্তি দুনিয়ায় হাঁটেননি।", "BanglaNMT", "spaCy"],
-        ["তিনজনের কেউই বাবার পথ ধরে প্রযুক্তি দুনিয়ায় হাঁটেননি।", "MyMemory", "spaCy"],
-        ["বিশ্বের আরও একটি সেরা ক্লাব।", "Google", "Flair"]
+        [
+            "বাংলাদেশ দক্ষিণ এশিয়ার একটি সার্বভৌম রাষ্ট্র।", 
+            "Google", 
+            "Neulab-AwesomeAlign (Bn-En-0.5M)", 
+            "NLTK"
+        ],
+        [
+            "বাংলাদেশের সংবিধানিক নাম কি?", 
+            "Google", 
+            "Google-mBERT (Base-Multilingual)",
+            "spaCy"
+        ],
+        [
+            "বাংলাদেশের সাংবিধানিক নাম গণপ্রজাতন্ত্রী বাংলাদেশ।", 
+            "Google", 
+            "Google-mBERT (Base-Multilingual)",
+            "TextBlob"
+        ],
+        [
+            "তিনজনের কেউই বাবার পথ ধরে প্রযুক্তি দুনিয়ায় হাঁটেননি।", 
+            "Google", 
+            "Neulab-AwesomeAlign (Bn-En-0.5M)", 
+            "spaCy"
+        ],
+        [
+            "তিনজনের কেউই বাবার পথ ধরে প্রযুক্তি দুনিয়ায় হাঁটেননি।", 
+            "BanglaNMT",
+            "Google-mBERT (Base-Multilingual)", 
+            "spaCy"
+        ],
+        [
+            "তিনজনের কেউই বাবার পথ ধরে প্রযুক্তি দুনিয়ায় হাঁটেননি।", 
+            "MyMemory",
+            "Google-mBERT (Base-Multilingual)", 
+            "spaCy"
+        ],
+        [
+            "বিশ্বের আরও একটি সেরা ক্লাব।", 
+            "Google", 
+            "Neulab-AwesomeAlign (Bn-En-0.5M)", 
+            "Flair"
+        ]
 
     ], inputs)