Update core/refiner/simple_refiner.py
Browse files
core/refiner/simple_refiner.py
CHANGED
@@ -22,7 +22,7 @@ class SimpleRefiner(BaseRefiner):
|
|
22 |
if not isinstance(image_paths, list):
|
23 |
image_paths = [image_paths]
|
24 |
for image_path in image_paths:
|
25 |
-
user_context.append({"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{encode_image(image_path.name)}"}})
|
26 |
context = [{"role": "system", "content": self.sys_prompt}] + memory + user_context
|
27 |
|
28 |
respond_task = asyncio.create_task(self.model.respond_async(context))
|
|
|
22 |
if not isinstance(image_paths, list):
|
23 |
image_paths = [image_paths]
|
24 |
for image_path in image_paths:
|
25 |
+
user_context[0]["content"].append({"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{encode_image(image_path.name)}"}})
|
26 |
context = [{"role": "system", "content": self.sys_prompt}] + memory + user_context
|
27 |
|
28 |
respond_task = asyncio.create_task(self.model.respond_async(context))
|