Views
No views yet
1from transformers import pipeline, AutoTokenizer, GPT2LMHeadModel, AutoConfig
2tokenizer = AutoTokenizer.from_pretrained('Arnavaz/gpt2-arnavaz-beta')
3model = GPT2LMHeadModel.from_pretrained('Arnavaz/gpt2-arnavaz-beta')
4config = AutoConfig.from_pretrained('Arnavaz/gpt2-arnavaz-beta', max_length=512)
5generator = pipeline('text-generation', model, tokenizer=tokenizer, config=config)
6
7def getEloquent(ineloquent):
8 result = generator(f"[BOS]{ineloquent}[SEP]")[0]['generated_text']
9 return result[result.find('[SEP]')+5:]
10
11sample = getEloquent('استفاده از کاغذ پاپیروس برای نوشتن کتاب از حدود دو هزار سال قبل از میلاد در مصر رایج شد.')