metadata
tags:
- vision
- image-to-image
- endpoints-template
inference: false
pipeline_tag: image-to-image
base_model: timbrooks/instruct-pix2pix
library_name: generic
timbrooks/instruct-pix2pix to deploy with Inference Endpoints
Expected payload:
def predict(path_to_image, prompt):
with open(path_to_image, "rb") as i:
b64 = base64.b64encode(i.read()).decode()
payload = {
"inputs": b64,
"parameters": {
"prompt": prompt
}
}
response = r.post(
ENDPOINT_URL, json=payload, headers={"Content-Type": "application/json"}
)
return response.json()
Call it with:
resp = predict(
path_to_image="car.png",
prompt="make the car green"
)
img = Image.open(BytesIO(base64.b64decode(resp)))