API_URL = "
https://api-inference.huggingface.co/models/Kvikontent/midjourney-v6"
headers = {"Authorization": "Bearer HUGGINGFACE_API_TOKEN"}
def query(payload):
response = requests.post(API_URL, headers=headers, json=payload)
return response.content
image_bytes = query({
"inputs": "Astronaut riding a horse",
})