app.py

import streamlit as st
from transformers import GPT2LMHeadModel, GPT2Tokenizer
st.title("자동 일기 생성기")

keywords = st.text_input("5개의 키워드를 입력하세요 (쉼표로 구분)", "")
keyword_list = [kw.strip() for kw in keywords.split(",")]

if len(keyword_list) == 5 and st.button("일기 쓰기"):
    # 모델 및 토크나이저 로드
    model = GPT2LMHeadModel.from_pretrained("gpt2")
    tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
    
    # 키워드 기반 fine-tuning
    input_ids = tokenizer.encode(" ".join(keyword_list), return_tensors="pt")
    output = model.generate(input_ids, max_length=500, num_return_sequences=1, do_sample=True, top_k=50, top_p=0.95, num_beams=5)
    
    # 생성된 일기 출력
    diary = tokenizer.decode(output[0], skip_special_tokens=True)
    st.write(diary)