Hugging Face's logo

Spaces:
Didier
/
Mistral_Small_AutoRound
Sleeping

App Files Community
Didier commited on
Commit
f870519
·
verified ·
1 Parent(s): a6f0c4b

Update module_vision.py

Browse files
Files changed (1) hide show
  1. module_vision.py +1 -1
module_vision.py CHANGED
@@ -11,7 +11,7 @@ import vlm
11
  def process(message, history):
12
  """Generate the model response given message and history
13
  """
14
- messages = history
15
  messages.append({'role': 'user', 'content': message})
16
  yield from vlm.stream_response(messages)
17
 
 
11
  def process(message, history):
12
  """Generate the model response given message and history
13
  """
14
+ messages = vlm.build_messages(message, history)
15
  messages.append({'role': 'user', 'content': message})
16
  yield from vlm.stream_response(messages)
17