Spaces:
Runtime error
Runtime error
musfiqdehan
commited on
Commit
•
db29d97
1
Parent(s):
f19f4b7
Add SagorSarker-BanglaBERT (Base) model to alignment_mappers.py
Browse files
helper/alignment_mappers.py
CHANGED
@@ -22,6 +22,8 @@ def select_model(model_name):
|
|
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 |
|
|
|
22 |
model_name="musfiqdehan/bn-en-word-aligner"
|
23 |
elif model_name == "BUET-BanglaBERT (Large)":
|
24 |
model_name="csebuetnlp/banglabert_large"
|
25 |
+
elif model_name == "SagorSarker-BanglaBERT (Base)":
|
26 |
+
model_name="sagorsarker/bangla-bert-base"
|
27 |
|
28 |
return model_name
|
29 |
|