ericbanzuzi commited on
Commit
7b8953c
·
1 Parent(s): 4ea81e1
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ llm = Llama.from_pretrained(
8
  )
9
 
10
  def predict(message, history):
11
- messages = [{"role": "system", "content": "You are a helpful assistant."}]
12
  for user_message, bot_message in history:
13
  if user_message:
14
  messages.append({"role": "user", "content": user_message})
@@ -26,7 +26,7 @@ def predict(message, history):
26
  response += part
27
  yield response
28
 
29
- demo = gr.ChatInterface(predict)
30
 
31
  if __name__ == "__main__":
32
  demo.launch()
 
8
  )
9
 
10
  def predict(message, history):
11
+ messages = [{"role": "system", "content": "You are a helpful assistant who answers questions in a concise but thorough way. Prioritize clarity and usefulness in all interactions."}]
12
  for user_message, bot_message in history:
13
  if user_message:
14
  messages.append({"role": "user", "content": user_message})
 
26
  response += part
27
  yield response
28
 
29
+ demo = gr.ChatInterface(predict, theme="Shivi/calm_seafoam")
30
 
31
  if __name__ == "__main__":
32
  demo.launch()