Views
No views yet
This is part of the Flax/Jax Community Week, organized by HuggingFace and TPU usage sponsored by Google.
1from transformers import pipeline, AutoTokenizer, GPT2LMHeadModel
2tokenizer = AutoTokenizer.from_pretrained('flax-community/gpt2-medium-persian')
3model = GPT2LMHeadModel.from_pretrained('flax-community/gpt2-medium-persian')
4generator = pipeline('text-generation', model, tokenizer=tokenizer, config={'max_length':100})
5generated_text = generator('در یک اتفاق شگفت انگیز، پژوهشگران')