arxyzan's picture
Update README.md
01e1f1d verified
|
raw
history blame
521 Bytes
---
language:
- fa
library_name: hezar
tags:
- fill-mask
- hezar
pipeline_tag: fill-mask
base_model:
- hezarai/distilbert-base-fa
---
A DistilBERT model for mask filling. The base DistilBERT model is initialized from https://huggingface.co/HooshvareLab/distilbert-fa-zwnj-base.
## Usage
```
pip install hezar
```
```python
from hezar.models import Model
model = Model.load("hezarai/distilbert-fa-mask-filling")
inputs = ["ایشالا از [MASK] شروع میکنم"]
outputs = model.predict(inputs)
print(outputs)
```