Upload app.py

app.py

@@ -0,0 +1,24 @@
+import streamlit as st
+from transformers import GPT2LMHeadModel, GPT2Tokenizer
+
+def app():
+    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)
+
+if __name__ == "__main__":
+    app()