Artples commited on
Commit
84029e4
·
verified ·
1 Parent(s): 130e4a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 += [{"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
 
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