| #!/usr/bin/env python3 | |
| import requests | |
| import io | |
| from PIL import Image | |
| API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1" | |
| headers = {"Authorization": "Bearer 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"} | |
| def query(payload): | |
| response = requests.post(API_URL, headers=headers, json=payload) | |
| return response.content | |
| image_bytes = query({ | |
| "inputs": "Astronaut riding a horse", | |
| }) | |
| # You can access the image with PIL.Image for example | |
| image = Image.open(io.BytesIO(image_bytes)) | |
| import ipdb; ipdb.set_trace() | |