Seg2Sat-endpoint / test_endpoint.py
rgres's picture
Upload folder using huggingface_hub
1273ba8
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")