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
|