freddyaboulton HF Staff commited on
Commit
1700e55
·
verified ·
1 Parent(s): 1255de4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -209,8 +209,9 @@ def generate(data: WebRTCData, history: list[dict], system_prompt: str = "", max
209
  files.append(image)
210
  message = {
211
  "text": data.textbox,
212
- "files": [],
213
  }
 
214
  new_message = {"role": "assistant", "content": ""}
215
  for output in _generate(message, history, system_prompt, max_new_tokens):
216
  new_message["content"] += output
 
209
  files.append(image)
210
  message = {
211
  "text": data.textbox,
212
+ "files": files,
213
  }
214
+ history.append({"role": "user", "content": data.textbox})
215
  new_message = {"role": "assistant", "content": ""}
216
  for output in _generate(message, history, system_prompt, max_new_tokens):
217
  new_message["content"] += output