Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
+# 모델과 토크나이저를 전역 변수로 미리 로드
+model = GPT2LMHeadModel.from_pretrained("gpt2")
+tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
+
 def generate_diary(keywords):
-    # 모델 및 토크나이저 로드
-    model = GPT2LMHeadModel.from_pretrained("gpt2")
-    tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
-    
     # 키워드 기반 fine-tuning
     input_ids = tokenizer.encode(" ".join(keywords), 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)