cochi1706 commited on
Commit
26e874e
·
1 Parent(s): db09be5

Add default system message in Vietnamese for chatbot responses and clarify instruction for model output language. This ensures consistent responses in the desired language when no system message is provided.

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -57,9 +57,15 @@ def respond(
57
  # Chuẩn bị messages từ history và message hiện tại
58
  messages = []
59
 
60
- # Thêm system message nếu có
61
  if system_message:
62
  messages.append({"role": "system", "content": system_message})
 
 
 
 
 
 
63
 
64
  # Thêm history (đã có format đúng)
65
  messages.extend(history)
@@ -75,6 +81,10 @@ def respond(
75
  add_generation_prompt=True
76
  )
77
 
 
 
 
 
78
  # Sử dụng pipeline để generate text - chỉ dùng max_new_tokens
79
  generated = text_generator(
80
  prompt,
 
57
  # Chuẩn bị messages từ history và message hiện tại
58
  messages = []
59
 
60
+ # Thêm system message - mặc định bằng tiếng Việt nếu không
61
  if system_message:
62
  messages.append({"role": "system", "content": system_message})
63
+ else:
64
+ # System message mặc định bằng tiếng Việt để model trả lời bằng tiếng Việt
65
+ messages.append({
66
+ "role": "system",
67
+ "content": "Bạn là một trợ lý lập trình hữu ích. Hãy trả lời bằng tiếng Việt. Cung cấp các giải pháp code rõ ràng, ngắn gọn và chính xác cùng với giải thích."
68
+ })
69
 
70
  # Thêm history (đã có format đúng)
71
  messages.extend(history)
 
81
  add_generation_prompt=True
82
  )
83
 
84
+ # Thêm instruction rõ ràng để model trả lời bằng tiếng Việt
85
+ # (nếu model vẫn có xu hướng trả lời tiếng Anh)
86
+ # prompt += "\n\nLưu ý: Hãy trả lời bằng tiếng Việt."
87
+
88
  # Sử dụng pipeline để generate text - chỉ dùng max_new_tokens
89
  generated = text_generator(
90
  prompt,