Spaces:
Runtime error
Runtime error
musfiqdehan
commited on
Commit
•
f19f4b7
1
Parent(s):
b040348
Update translation model choices
Browse files- app.py +1 -1
- 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-
|
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-
|
24 |
-
model_name="csebuetnlp/
|
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 |
|