File size: 501 Bytes
30fff05 e307ddc 30fff05 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
import sys
import pprint
from transformers import pipeline, AutoTokenizer, RobertaForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("./")
model = RobertaForMaskedLM.from_pretrained("./", from_flax=True)
nlp = pipeline("fill-mask", model=model, tokenizer=tokenizer)
# masked_input = "माझा नाव <mask> आहे"
# masked_input = "वाढदिवसाच्या <mask> शुभेच्छा!"
masked_input = sys.argv[1]
print(masked_input)
pprint.pprint(nlp(masked_input))
|