import os
import requests
import json
def generate_image():
url = "https://api.gitcode.com/api/v5/images/generations"
headers = {
"Authorization": f"Bearer {os.environ['ACCESS_TOKEN']}",
"Content-Type": "application/json"
}
data = {
"model": "Qwen/Qwen-Image",
"prompt": "生成一只小猫"
}
response = requests.post(url, headers=headers, json=data)
result = response.json()
if result.get("status") == "success":
image_url = result["images"][0]
print(f"Image URL: {image_url}")
return image_url
else:
print("Image generation failed")
return None
generate_image()