davidberenstein1957 HF staff commited on
Commit
7ccc989
1 Parent(s): b7d4f1f

Add system prompts

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def generate(
30
  top_k: int = 40,
31
  repetition_penalty: float = 1.2,
32
  ) -> Iterator[str]:
33
- conversation = []
34
  for user, assistant in chat_history:
35
  conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
36
  conversation.append({"role": "user", "content": message})
 
30
  top_k: int = 40,
31
  repetition_penalty: float = 1.2,
32
  ) -> Iterator[str]:
33
+ conversation = [{"role": "system", "content": "be concise"}]
34
  for user, assistant in chat_history:
35
  conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
36
  conversation.append({"role": "user", "content": message})