vilarin commited on
Commit
9986148
1 Parent(s): 29c4e7f

Update app.py

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