cutiee82 commited on
Commit
5720b37
·
verified ·
1 Parent(s): 91b7faf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -6,13 +6,13 @@ SYSTEM_PROMPT = os.environ["SYSTEM_PROMPT"]
6
  client = InferenceClient("HuggingFaceTB/SmolLM2-1.7B-Instruct")
7
 
8
  def chat(message, history):
9
- msg = [{"role": "system", "content": SYSTEM_PROMPT}]
10
- msg.extend(history)
11
- msg.append({"role": "user", "content": message})
12
 
13
  result = ""
14
  for chunk in client.chat.completions.create(
15
- messages=msg,
16
  max_tokens=1024,
17
  stream=True
18
  ):
 
6
  client = InferenceClient("HuggingFaceTB/SmolLM2-1.7B-Instruct")
7
 
8
  def chat(message, history):
9
+ messages = [{"role": "system", "content": SYSTEM_PROMPT}]
10
+ messages.extend(history)
11
+ messages.append({"role": "user", "content": message})
12
 
13
  result = ""
14
  for chunk in client.chat.completions.create(
15
+ messages,
16
  max_tokens=1024,
17
  stream=True
18
  ):