You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

38 lines
1.6 KiB

2 weeks ago
from openai import OpenAI
client = OpenAI(
api_key='sk-f6da0c787eff4b0389e4ad03a35a911f',
base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
)
2 weeks ago
2 weeks ago
prompt = "请提取图片中的试题"
2 weeks ago
completion = client.chat.completions.create(
model="qwen-vl-ocr-latest",
messages=[
{
"role": "user",
"content": [
{
"type": "image_url",
2 weeks ago
"image_url": "https://pic1.zhimg.com/v2-c3d7f060bbf3f7c122319350044d8888_1440w.jpg",
2 weeks ago
# 输入图像的最小像素阈值小于该值图像会按原比例放大直到总像素大于min_pixels
"min_pixels": 28 * 28 * 4,
# 输入图像的最大像素阈值超过该值图像会按原比例缩小直到总像素低于max_pixels
"max_pixels": 28 * 28 * 8192
},
2 weeks ago
# qwen-vl-ocr-latest支持在以下text字段中传入Prompt若未传入则会使用默认的PromptPlease output only the text content from the image without any additional descriptions or formatting.
# 如调用qwen-vl-ocr-1028模型会使用固定PromptRead all the text in the image.不支持用户在text中传入自定义Prompt
2 weeks ago
{"type": "text",
2 weeks ago
"text": prompt},
2 weeks ago
]
}
])
2 weeks ago
print(completion.choices[0].message.content)
# 将返回的内容保存到 2、识别出结果.md 中
2 weeks ago
with open('Res/2、识别出结果.md', 'w', encoding='utf-8') as f:
2 weeks ago
f.write(completion.choices[0].message.content)
2 weeks ago
print("保存成功!")