File size: 277 Bytes
b3fc01c
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
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'])