musfiqdehan's picture
Add SagorSarker-BanglaBERT as a model option
eeb5e90
raw
history blame
5.26 kB
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, model_name, tagger):
"""
Bangla PoS Tagger
"""
src = space_punc(src)
tgt_base, tgt = select_translator(src, translator)
model_name = select_model(model_name)
result, pos_accuracy = select_pos_tagger(src, tgt, model_name, tagger)
return tgt_base, result, pos_accuracy
# Define the Gradio interface
# demo = gr.Interface(
# fn=bn_postagger,
# inputs=[
# gr.Textbox(label="Enter Bangla Sentence", placeholder="বাংলা বাক্য লিখুন"),
# gr.Dropdown(["Google", "BanglaNMT", "MyMemory"], label="Select a Translator"),
# gr.Dropdown(["spaCy", "NLTK", "Flair", "TextBlob"], label="Select a PoS Tagger")
# ],
# outputs= [
# gr.Textbox(label="English Translation"),
# RichTextbox(label="PoS Tags"),
# gr.Textbox(label="Overall PoS Tagging Accuracy")
# ],
# live=False,
# title="Bangla PoS Taggers",
# theme='',
# examples=[
# ["বাংলাদেশ দক্ষিণ এশিয়ার একটি সার্বভৌম রাষ্ট্র।"],
# ["বাংলাদেশের সংবিধানিক নাম কি?"],
# ["বাংলাদেশের সাংবিধানিক নাম গণপ্রজাতন্ত্রী বাংলাদেশ।"],
# ["তিনজনের কেউই বাবার পথ ধরে প্রযুক্তি দুনিয়ায় হাঁটেননি।"],
# ["বিশ্বের আরও একটি সেরা ক্লাব।"]
# ]
# )
with gr.Blocks(css="styles.css") as demo:
gr.HTML("<h1>Bangla PoS Taggers</h1>")
gr.HTML("<p>Parts of Speech (PoS) Tagging of Bangla Sentence using Bangla-English <strong>Word Alignment</strong></p>")
with gr.Row():
with gr.Column():
inputs = [
gr.Textbox(
label="Enter Bangla Sentence",
placeholder="বাংলা বাক্য লিখুন"
),
gr.Dropdown(
choices=["Google", "BanglaNMT", "MyMemory"],
label="Select a Translator"
),
gr.Dropdown(
choices=["Google-mBERT (Base-Multilingual)", "Neulab-AwesomeAlign (Bn-En-0.5M)", "BUET-BanglaBERT (Large)", "SagorSarker-BanglaBERT (Base)"],
label="Select a Model"
),
gr.Dropdown(
choices=["spaCy", "NLTK", "Flair", "TextBlob"],
label="Select a PoS Tagger"
)
]
btn = gr.Button(value="Submit", elem_classes="mybtn")
gr.ClearButton(inputs)
with gr.Column():
outputs = [
gr.Textbox(label="English Translation"),
RichTextbox(label="PoS Tags"),
gr.Textbox(label="PoS Tagging Accuracy (Based on Unknown(UNK) Tags)")
]
btn.click(bn_postagger, inputs, outputs)
gr.Examples([
[
"বাংলাদেশ দক্ষিণ এশিয়ার একটি সার্বভৌম রাষ্ট্র।",
"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)
# Launch the Gradio app
if __name__ == "__main__":
demo.launch()