from transformers import pipeline

# Model load කරන්න
generator = pipeline("text-generation", 
                     model="gpt2")

# Text generate කරන්න
result = generator("Sri Lanka is", 
                   max_length=50)
print(result[0]['generated_text'])