Update app.py

app.py

@@ -13,7 +13,7 @@ def generate_answer(input_text):
     # input_ids = tokenizer(input_text, max_length=700, return_tensors="pt").input_ids
     
     # 모델 추론
-    output_ids = model.generate(input_ids, max_length=1024)
+    output_ids = model.generate(input_ids, top_k=10, max_length=1024)
     # output_ids = model.generate(input_ids, num_beams=10, top_k=10, max_length=1024)
     
     # 모델 출력을 텍스트로 디코딩