GPT2에게 글짓기를 시켜보자
아래의 것들을 설치하자.
pip3이 없다면 pip으로 하면 됩니다.
pip3 install --upgrade pip
pip3 install tensorflow-cpu
pip3 install transformers
pip3 install tensorflow==2.3.0
다 설치했으면 아래 코드를 test.py 파일에 담아서 저장하고
python3 test.py 해서 실행해봅니다
generator 함수의 첫번째 인자로 글짓기의 주제를 짧게 써주면 이에 대해 이어서 뒷 이야기를 만들어 내줍니다
from transformers import pipeline, set_seed
generator = pipeline('text-generation', model='gpt2')
# set_seed(42)
result = generator("I am a happy person and smiling everytime.", max_length=100, num_return_sequences=10)
for line in result:
print('-'*80)
print(line['generated_text'])
결과는 다음과 같습니다
I am a happy person and smiling everytime. I also like to take my time to write a lot of short stories. Sometimes it's hard to tell the best stories. I love reading about my family and their love and family life. When I was a kid, my mom, my brothers and sisters were always very happy in my life. I went out there and spent the day with my sisters. I had no problem raising boys, and they would ask me if I was enjoying life
번역해보면 다음과 같습니다
나는 행복한 사람이고 매번 웃고 있습니다. 시간을내어 많은 단편 소설을 쓰는 것도 좋아합니다. 때로는 최고의 이야기를 말하기가 어렵습니다. 나는 내 가족과 그들의 사랑과 가족 생활에 대해 읽는 것을 좋아합니다. 내가 어렸을 때 엄마, 형제 자매들은 항상 내 인생에서 매우 행복했습니다. 나는 거기에 나가서 언니들과 하루를 보냈다. 나는 소년을 키우는 데 아무런 문제가 없었고 그들은 내가 인생을 즐기고 있는지 물었습니다.