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)
|