qnguyen3 commited on
Commit
2a565ab
1 Parent(s): 60ca437

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODELS)
47
  def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
48
  print(f'message is - {message}')
49
  print(f'history is - {history}')
50
- conversation = []
51
  for prompt, answer in history:
52
  conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
53
  conversation.append({"role": "user", "content": message})
 
47
  def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
48
  print(f'message is - {message}')
49
  print(f'history is - {history}')
50
+ conversation = [{"role": "system", "content": 'You are a helpful assistant. Answer in proper Vietnamese.'}]
51
  for prompt, answer in history:
52
  conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
53
  conversation.append({"role": "user", "content": message})