schuler commited on
Commit
e16be07
·
verified ·
1 Parent(s): 8deab94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=256):
88
  for message in chat_history:
89
  role = "<|assistant|>" if message['role'] == 'assistant' else "<|user|>"
90
  prompt += f"\n{role}\n{message['content']}\n"
91
- prompt += f"\n<|user|>\n {user_text}\n<|assistant|>\"
92
 
93
  # Generate the response
94
  response_output = generator(
 
88
  for message in chat_history:
89
  role = "<|assistant|>" if message['role'] == 'assistant' else "<|user|>"
90
  prompt += f"\n{role}\n{message['content']}\n"
91
+ prompt += f"\n<|user|>\n{user_text}\n<|assistant|>\n"
92
 
93
  # Generate the response
94
  response_output = generator(