百度文心 ERNIE-ViLG 的 API Key 和图片链接使用指南

文心AI作画API的强大功能

文心AI作画是一款基于百度文心 ERNIE-ViLG 模型的跨模态AI绘画工具，旨在通过自然语言描述生成丰富多样的图像。ERNIE-ViLG 是一个中文跨模态生成模型，拥有100亿参数规模，能够通过自回归算法将图像与文本生成统一建模。

在艺术创作、虚拟现实及AI辅助设计等领域，文心AI作画展现了其强大的创造力和实用性。该工具不仅能激发用户的想象力，还为多种行业提供了高效便捷的图像解决方案。

ERNIE-ViLG生成的不同风格的图像

如何获取并使用API Key

要使用文心AI作画，首先需要在百度智能云平台上注册账号并获取API Key和Secret Key。这些凭证是访问API的必要条件。

注册百度智能云账号

前往百度智能云网站并进行注册。
登录后，在控制台选择“产品服务”中的“人工智能”模块。
选择“智能创作平台”项目，完成实名认证后领取免费的接口资源。

注册界面

创建应用并获取API Key

在“应用列表”中点击“创建应用”，选择文本-图像生成应用模块。
创建成功后，将获得API Key和Secret Key，用于后续API请求。

API Key获取

文心AI作画API的使用方法

文心AI作画API主要通过提交请求和查询结果两个接口实现图像生成。

提交请求接口

提交请求接口用于根据用户输入的文本生成图像任务ID。请求信息包括文本描述、图片分辨率及风格参数。

import requests
import json

API_KEY = "你的API Key"
SECRET_KEY = "你的Secret Key"

url = "https://aip.baidubce.com/rpc/2.0/ernievilg/v1/txt2img?access_token=" + get_access_token()

payload = json.dumps({
    "text": "中国山水画",
    "resolution": "1024*1024",
    "style": "古风",
    "num": 2
})
headers = {
    'Content-Type': 'application/json',
    'Accept': 'application/json'
}

response = requests.request("POST", url, headers=headers, data=payload)
print(response.text)

查询结果接口

查询结果接口用于获取图像生成状态及最终图片链接。通过任务ID查询，生成图像的状态和地址将被返回。

import requests
import json

url = "https://aip.baidubce.com/rpc/2.0/ernievilg/v1/getImg?access_token=" + get_access_token()

payload = json.dumps({
    "taskId": "提交请求返回的taskId"
})
headers = {
    'Content-Type': 'application/json',
    'Accept': 'application/json'
}

response = requests.request("POST", url, headers=headers, data=payload)
print(response.text)

效果展示

使用文心speed模型

文心speed模型提供了一种快速调用百度AI作画API的方法。用户可以通过命令行或Python代码实现快速调用。

命令行测试

通过命令行调用API，需先获取access_token，并在API调用中使用。

curl 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=[API Key]&client_secret=[Secret Key]'
curl -XPOST  'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie_speed?access_token=[access_token]' -d '{"messages": [{"role":"user","content":"介绍一下北京"}]}' | iconv -f utf-8 -t utf-8

Python调用

通过Python脚本可以实现单轮或多轮对话的API调用，将access_token填入脚本中即可。

import requests
import json

def get_access_token():
    url = "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=[API Key]&client_secret=[Secret Key]"
    response = requests.post(url).json()
    return response.get("access_token")

url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie_speed?access_token=" + get_access_token()
payload = json.dumps({"messages": [{"role": "user", "content": "介绍一下北京"}]})
headers = {'Content-Type': 'application/json'}
response = requests.request("POST", url, headers=headers, data=payload)
print(response.text)