diffusers/tools

1#!/usr/bin/env python32import requests3import io4from PIL import Image5 6API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1"7headers = {"Authorization": "Bearer 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"}8 9def query(payload):10	response = requests.post(API_URL, headers=headers, json=payload)11	return response.content12image_bytes = query({13	"inputs": "Astronaut riding a horse",14})15# You can access the image with PIL.Image for example16image = Image.open(io.BytesIO(image_bytes))17import ipdb; ipdb.set_trace()18