Update README.md

README.md

@@ -6,6 +6,8 @@ tags:
 - fill-mask
 - hezar
 pipeline_tag: fill-mask
+base_model:
+- hezarai/bert-base-fa
 ---
 A BERT model for mask filling. The base BERT model is initialized from https://huggingface.co/HooshvareLab/bert-fa-zwnj-base.
 
@@ -20,4 +22,4 @@ model = Model.load("hezarai/bert-fa-mask-filling")
 inputs = ["ایشالا از [MASK] شروع میکنم"]
 outputs = model.predict(inputs)
 print(outputs)
-```
+```