|
--- |
|
license: openrail |
|
tags: |
|
- stable-diffusion |
|
- stable-diffusion-diffusers |
|
- controlnet |
|
inference: true |
|
--- |
|
|
|
# Inference Endpoint for [Seg2Sat](https://huggingface.co/rgres/Seg2Sat-sd-controlnet) using [runwayml/stable-diffusion-v1-5](https://huggingface.co/stabilityai/stable-diffusion-2-1-base) |
|
|
|
The code from the project can be found here: https://github.com/RubenGres |
|
Inference endpoint for Seg2Map used on the demo available on rubengr.es/Seg2Sat |
|
|
|
|
|
```python |
|
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") |
|
|
|
``` |