NLP
Collection
NLP models, datasets, etc.
•
23 items
•
Updated
•
3
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)
Inference API (serverless) does not yet support hezar models for this pipeline type.