musfiqdehan commited on
Commit
f19f4b7
1 Parent(s): b040348

Update translation model choices

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. helper/alignment_mappers.py +2 -2
app.py CHANGED
@@ -65,7 +65,7 @@ with gr.Blocks(css="styles.css") as demo:
65
  label="Select a Translator"
66
  ),
67
  gr.Dropdown(
68
- choices=["Google-mBERT (Base-Multilingual)", "Neulab-AwesomeAlign (Bn-En-0.5M)", "BUET-BanglaT5"],
69
  label="Select a Model"
70
  ),
71
  gr.Dropdown(
 
65
  label="Select a Translator"
66
  ),
67
  gr.Dropdown(
68
+ choices=["Google-mBERT (Base-Multilingual)", "Neulab-AwesomeAlign (Bn-En-0.5M)", "BUET-BanglaBERT (Large)"],
69
  label="Select a Model"
70
  ),
71
  gr.Dropdown(
helper/alignment_mappers.py CHANGED
@@ -20,8 +20,8 @@ def select_model(model_name):
20
  model_name="bert-base-multilingual-cased"
21
  elif model_name == "Neulab-AwesomeAlign (Bn-En-0.5M)":
22
  model_name="musfiqdehan/bn-en-word-aligner"
23
- elif model_name == "BUET-BanglaT5":
24
- model_name="csebuetnlp/banglat5"
25
 
26
  return model_name
27
 
 
20
  model_name="bert-base-multilingual-cased"
21
  elif model_name == "Neulab-AwesomeAlign (Bn-En-0.5M)":
22
  model_name="musfiqdehan/bn-en-word-aligner"
23
+ elif model_name == "BUET-BanglaBERT (Large)":
24
+ model_name="csebuetnlp/banglabert_large"
25
 
26
  return model_name
27