lindsay-qu commited on
Commit
7721116
·
verified ·
1 Parent(s): bee9e0d

Update core/refiner/simple_refiner.py

Browse files
Files changed (1) hide show
  1. core/refiner/simple_refiner.py +1 -1
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))