JuanjoSG5 commited on
Commit
e0b4034
·
1 Parent(s): d0a1746

test: assertion error in stream

Browse files
Files changed (1) hide show
  1. 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
- messages.append(HumanMessage(content={"type":"text","text":message}))
73
- messages.append(HumanMessage(content={"type":"image_url","image_url":encoded}))
 
 
 
 
 
 
 
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)