JuanjoSG5
commited on
Commit
·
e0b4034
1
Parent(s):
d0a1746
test: assertion error in stream
Browse files- gradio_interface/app.py +10 -2
gradio_interface/app.py
CHANGED
|
@@ -68,9 +68,17 @@ def generate_response(message, chat_history, image):
|
|
| 68 |
messages.append(HumanMessage(content=content))
|
| 69 |
else:
|
| 70 |
messages.append(AIMessage(content=content))
|
|
|
|
| 71 |
encoded = encode_image_to_base64(image)
|
| 72 |
-
|
| 73 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 74 |
|
| 75 |
try:
|
| 76 |
stream_iter = llm.stream(messages)
|
|
|
|
| 68 |
messages.append(HumanMessage(content=content))
|
| 69 |
else:
|
| 70 |
messages.append(AIMessage(content=content))
|
| 71 |
+
|
| 72 |
encoded = encode_image_to_base64(image)
|
| 73 |
+
|
| 74 |
+
# Create a properly formatted multimodal content list
|
| 75 |
+
multimodal_content = [
|
| 76 |
+
{"type": "text", "text": message},
|
| 77 |
+
{"type": "image_url", "image_url": {"url": encoded}}
|
| 78 |
+
]
|
| 79 |
+
|
| 80 |
+
# Append a single HumanMessage with the multimodal content
|
| 81 |
+
messages.append(HumanMessage(content=multimodal_content))
|
| 82 |
|
| 83 |
try:
|
| 84 |
stream_iter = llm.stream(messages)
|