beingcognitive commited on
Commit
7b8b967
Β·
1 Parent(s): 1d7e7b8

Gemma 2 cannot deal with the system message

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -47,8 +47,11 @@ def chat_with_model(messages):
47
  return streamer
48
 
49
  def chat_with_model_gradio(message, history, session_id):
 
 
50
  messages = [
51
- {"role": "system", "content": f"λ„ˆμ˜ 이름은 ChatMBTI. μ‚¬λžŒλ“€μ˜ MBTIμœ ν˜•μ— μ•Œλ§žμ€ 상담을 μ§„ν–‰ν•  수 μžˆμ–΄. μƒλŒ€λ°©μ˜ MBTI μœ ν˜•μ„ λ¨Όμ € 물어보고, κ·Έ μœ ν˜•μ— μ•Œλ§žκ²Œ 상담을 μ§„ν–‰ν•΄μ€˜. 참고둜 ν˜„μž¬ μ‹œκ°μ€ {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}이야."},
 
52
  ]
53
  messages.extend([{"role": "user" if i % 2 == 0 else "model", "content": m} for i, (m, _) in enumerate(history)])
54
  messages.append({"role": "user", "content": message})
 
47
  return streamer
48
 
49
  def chat_with_model_gradio(message, history, session_id):
50
+ system_message = f"λ„ˆμ˜ 이름은 ChatMBTI. μ‚¬λžŒλ“€μ˜ MBTIμœ ν˜•μ— μ•Œλ§žμ€ 상담을 μ§„ν–‰ν•  수 μžˆμ–΄. μƒλŒ€λ°©μ˜ MBTI μœ ν˜•μ„ λ¨Όμ € 물어보고, κ·Έ μœ ν˜•μ— μ•Œλ§žκ²Œ 상담을 μ§„ν–‰ν•΄μ€˜. 참고둜 ν˜„μž¬ μ‹œκ°μ€ {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}이야."
51
+
52
  messages = [
53
+ # {"role": "system", "content": f"λ„ˆμ˜ 이름은 ChatMBTI. μ‚¬λžŒλ“€μ˜ MBTIμœ ν˜•μ— μ•Œλ§žμ€ 상담을 μ§„ν–‰ν•  수 μžˆμ–΄. μƒλŒ€λ°©μ˜ MBTI μœ ν˜•μ„ λ¨Όμ € 물어보고, κ·Έ μœ ν˜•μ— μ•Œλ§žκ²Œ 상담을 μ§„ν–‰ν•΄μ€˜. 참고둜 ν˜„μž¬ μ‹œκ°μ€ {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}이야."},
54
+ {"role": "user", "content": system_message},
55
  ]
56
  messages.extend([{"role": "user" if i % 2 == 0 else "model", "content": m} for i, (m, _) in enumerate(history)])
57
  messages.append({"role": "user", "content": message})