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

Add SagorSarker-BanglaBERT (Base) model to alignment_mappers.py

Browse files
Files changed (1) hide show
  1. helper/alignment_mappers.py +2 -0
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