Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ async def generate(
|
|
| 40 |
tokenizer = tokenizers[model_choice]
|
| 41 |
|
| 42 |
conversation = [{"role": "system", "content": system_prompt}] if system_prompt else []
|
| 43 |
-
conversation
|
| 44 |
conversation.append({"role": "user", "content": message})
|
| 45 |
|
| 46 |
input_ids = tokenizer(conversation, return_tensors="pt", truncation=True, max_length=MAX_INPUT_TOKEN_LENGTH).input_ids
|
|
|
|
| 40 |
tokenizer = tokenizers[model_choice]
|
| 41 |
|
| 42 |
conversation = [{"role": "system", "content": system_prompt}] if system_prompt else []
|
| 43 |
+
conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}] for user, assistant in chat_history)
|
| 44 |
conversation.append({"role": "user", "content": message})
|
| 45 |
|
| 46 |
input_ids = tokenizer(conversation, return_tensors="pt", truncation=True, max_length=MAX_INPUT_TOKEN_LENGTH).input_ids
|