Update app.py

app.py

@@ -95,8 +95,12 @@ def generate(
         include_values=False,
         include_metadata=True
     )
-    
-    conversation.append({"role": "user", "content": results["matches"][0]["metadata"]["text"]+f"\n\n위 문맥을 참고하여 질문 '{message}'에 답하면?"})
+
+    results = [result for result in results["matches"] if result["score"] > 0.6]
+    if len(results) > 0:
+        message = results[0]["metadata"]["text"] + f"\n\n위 문맥을 참고하여 질문 '{message}'에 답하면?"
+        
+    conversation.append({"role": "user", "content": message })
     
     input_ids = tokenizer.apply_chat_template(conversation, return_tensors="pt", add_generation_prompt=True)
     if input_ids.shape[1] > MAX_INPUT_TOKEN_LENGTH: