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