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
|