packages/inference/src/snippets/templates/python/requests/imageToImage.jinja (15 lines of code) (raw):
with open("{{inputs.asObj.inputs}}", "rb") as image_file:
image_base_64 = base64.b64encode(image_file.read()).decode('utf-8')
def query(payload):
with open(payload["inputs"], "rb") as f:
img = f.read()
payload["inputs"] = base64.b64encode(img).decode("utf-8")
response = requests.post(API_URL, headers=headers, json=payload)
return response.content
image_bytes = query({
{{ providerInputs.asJsonString }}
})
# You can access the image with PIL.Image for example
import io
from PIL import Image
image = Image.open(io.BytesIO(image_bytes))