arxyzan's picture
Update README.md
01e1f1d verified
metadata
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
from hezar.models import Model

model = Model.load("hezarai/distilbert-fa-mask-filling")
inputs = ["ایشالا از [MASK] شروع میکنم"]
outputs = model.predict(inputs)
print(outputs)