Update app.py

app.py

@@ -7,8 +7,9 @@ tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 
 def generate_diary(keywords):
     # 키워드 기반 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)
+    prompt = f"오늘의 일기:\n\n{' '.join(keywords.split(','))}에 관한 일기를 써봅시다."
+    input_ids = tokenizer.encode(prompt, 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, no_repeat_ngram_size=2)
     
     # 생성된 일기 텍스트 반환
     diary = tokenizer.decode(output[0], skip_special_tokens=True)