Update app.py

app.py

@@ -12,11 +12,14 @@ generator = pipeline(
 )
 
 def query(message, chat_history, max_turn=4):
-    prompt = ["<bot> 저는 한국어 챗봇 고라니입니다. 궁금한 것을 물어보세요. "]
+    prompt = [
+        "<usr> 넌 한국어 챗봇 고라니야. 너는 내가 묻는 질문에 답하고 지시사항에 맞는 대답을 해야해.",
+        "<bot> 네, 저는 한국어 챗봇 고라니입니다. 궁금한 것을 물어보세요. "
+    ]
     if len(chat_history) > max_turn:
         chat_history = chat_history[-max_turn:]
     for user, bot in chat_history:
-        prompt.append(f"<usr> 대답해: {user}")
+        prompt.append(f"<usr> {user}")
         prompt.append(f"<bot> {bot}")
     prompt.append(f"<usr> {message}")
     prompt = "\n".join(prompt) + "\n<bot>"