| 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)) | |