pix2pix-instruct-IE / README.md
sergeipetrov's picture
Update README.md
6ec8056 verified
---
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:
```python
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:
```python
resp = predict(
path_to_image="car.png",
prompt="make the car green"
)
img = Image.open(BytesIO(base64.b64decode(resp)))
```