Update app.py
Browse files
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
|
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(
|