Update app.py
Browse files
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
|