NLP
Collection
NLP models, datasets, etc. • 20 items • Updated
• 4
A RoBERTa model for mask filling. The base RoBERTa model is initialized from https://huggingface.co/HooshvareLab/roberta-fa-zwnj-base.
pip install hezar
from hezar.models import Model
model = Model.load("hezarai/roberta-fa-mask-filling")
inputs = ["ایشالا از <mask> شروع میکنم"]
outputs = model.predict(inputs)
print(outputs)
Base model
hezarai/roberta-base-fa