vilarin commited on
Commit
4793190
1 Parent(s): da99c88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ def stream_chat(message, history: list, temperature: float, max_new_tokens: int)
66
  conversation.extend([{"role": "user", "content":""}])
67
  conversation[0]['content'] = f"<|image_1|>\n{prompt}"
68
  conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
69
- conversation.append({"role": "user", "content": message['text']})
70
 
71
  print(f"Conversation is -\n{conversation}")
72
  inputs = processor.tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
 
66
  conversation.extend([{"role": "user", "content":""}])
67
  conversation[0]['content'] = f"<|image_1|>\n{prompt}"
68
  conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
69
+ conversation.append({"role": "user", "content": message['text']})
70
 
71
  print(f"Conversation is -\n{conversation}")
72
  inputs = processor.tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)