ericbanzuzi commited on
Commit
bfb50b2
·
1 Parent(s): 50aecdb
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ llm = Llama.from_pretrained(
7
  )
8
 
9
  def predict(message, history):
10
- messages = [{"role": "system", "content": "You are a helpful assistant."}]
11
  for user_message, bot_message in history:
12
  if user_message:
13
  messages.append({"role": "user", "content": user_message})
@@ -25,7 +25,7 @@ def predict(message, history):
25
  response += part
26
  yield response
27
 
28
- demo = gr.ChatInterface(predict)
29
 
30
  if __name__ == "__main__":
31
  demo.launch()
 
7
  )
8
 
9
  def predict(message, history):
10
+ 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."}]
11
  for user_message, bot_message in history:
12
  if user_message:
13
  messages.append({"role": "user", "content": user_message})
 
25
  response += part
26
  yield response
27
 
28
+ demo = gr.ChatInterface(predict, theme="Shivi/calm_seafoam")
29
 
30
  if __name__ == "__main__":
31
  demo.launch()