zhong-zhang commited on
Commit
e2fb0ce
·
verified ·
1 Parent(s): 486dee7

Update README.md

Browse files

update vllm inference

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -197,7 +197,7 @@ def predict(text_prompt: str, image: Image.Image):
197
  messages = [
198
  {"role": "system", "content": SYSTEM_PROMPT},
199
  {"role": "user", "content": [
200
- {"type": "text", "text": f"<Question>{text_prompt}</Question>\n当前屏幕截图:"},
201
  {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{encode_image(image)}"}}
202
  ]}
203
  ]
 
197
  messages = [
198
  {"role": "system", "content": SYSTEM_PROMPT},
199
  {"role": "user", "content": [
200
+ {"type": "text", "text": f"<Question>{text_prompt}</Question>\n当前屏幕截图:(<image>./</image>)"},
201
  {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{encode_image(image)}"}}
202
  ]}
203
  ]