| import base64 | |
| import requests | |
| API_URL = "https://zqz606ggn85ysase.us-east-1.aws.endpoints.huggingface.cloud" | |
| def encode_image(image_path): | |
| with open(image_path, "rb") as i: | |
| b64 = base64.b64encode(i.read()) | |
| return b64.decode("utf-8") | |
| prompt = "aerial view of jardin princier, Toulouse. Flowers, flowers, garden" | |
| image = encode_image("handdrawn.png") | |
| headers = { | |
| "Accept": "image/png", | |
| "Content-Type": "application/json" | |
| } | |
| # test the handler | |
| def query(payload): | |
| response = requests.post(API_URL, headers=headers, json=payload) | |
| return response.content | |
| payload = { | |
| "inputs": prompt, | |
| "prompt": prompt, | |
| "image": image, | |
| "steps": 20, | |
| "seed": 999 | |
| } | |
| import json | |
| with open('payload.json', 'w') as f: | |
| json.dump(payload, f) | |
| image_bytes = query(payload) | |
| # You can access the image with PIL.Image for example | |
| import io | |
| from PIL import Image | |
| image = Image.open(io.BytesIO(image_bytes)) | |
| image.save("output.png") | |