schuler commited on
Commit
08b4c0f
·
verified ·
1 Parent(s): c97bd5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -135,7 +135,7 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=256, co
135
  for message in chat_history:
136
  role = "<|assistant|>" if message['role'] == 'assistant' else "<|user|>"
137
  prompt += f"{role}{message['content']}<|end|>"
138
- prompt += f"<|user|>{user_text}\n<|end|><|assistant|>"
139
 
140
  # Generate the response
141
  response_output = generator(
 
135
  for message in chat_history:
136
  role = "<|assistant|>" if message['role'] == 'assistant' else "<|user|>"
137
  prompt += f"{role}{message['content']}<|end|>"
138
+ prompt += f"<|user|>{user_text}<|end|><|assistant|>"
139
 
140
  # Generate the response
141
  response_output = generator(