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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -205,13 +205,15 @@ def generate(data: WebRTCData, history: list[dict], system_prompt: str = "", max
205
  files = []
206
  if data.audio is not None and data.audio[1].size > 0:
207
  files.append(save_audio_to_cache(data.audio[1], data.audio[0], format="mp3", cache_dir=get_upload_folder()))
208
- if image is None:
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
 
205
  files = []
206
  if data.audio is not None and data.audio[1].size > 0:
207
  files.append(save_audio_to_cache(data.audio[1], data.audio[0], format="mp3", cache_dir=get_upload_folder()))
208
+ if image is not None:
209
  files.append(image)
210
  message = {
211
  "text": data.textbox,
212
  "files": files,
213
  }
214
  history.append({"role": "user", "content": data.textbox})
215
+ print("history", history)
216
+ yield history
217
  new_message = {"role": "assistant", "content": ""}
218
  for output in _generate(message, history, system_prompt, max_new_tokens):
219
  new_message["content"] += output