sayakpaul's picture
sayakpaul HF staff
Update README.md
add5a0b verified
---
library_name: diffusers
tags:
- stable-diffusion-3.5
- diffusers
- text-to-image
- endpoints-template
inference: false
---
> [!IMPORTANT]
> This repo duplicates the [Stable Diffusion 3.5 Large](https://huggingface.co/stabilityai/stable-diffusion-3.5-large) weights for demonstration purposes and it doesn't own any credits to the model. So, please be mindful of that and respect the original license of the model.
```py
from handler import EndpointHandler
my_handler = EndpointHandler()
payload = {"inputs": {"prompt": "a dog waiting for its companion to come."}}
# test the handler
image = my_handler(payload)
image.save("image.png")
```
![](./image.png)
We can use this repo to deploy SD3.5 Large on [Inference Endpoints](https://huggingface.co/docs/inference-endpoints) as well.
```py
import json
import requests
import base64
from PIL import Image
from io import BytesIO
ENDPOINT_URL = ""
HF_TOKEN = ""
def decode_base64_image(image_string):
base64_image = base64.b64decode(image_string)
buffer = BytesIO(base64_image)
return Image.open(buffer).save("image.png")
def predict(prompt: str = "a dog waiting for its companion to come."):
payload = {"inputs": {"prompt": prompt}}
response = requests.post(ENDPOINT_URL, headers={"Authorization": f"Bearer {HF_TOKEN}"}, json=payload)
resp = response.json()
return decode_base64_image(resp)
prediction = predict(prompt="the first animal on the mars")
```
This is how the final image would look like:
![](./ie_image.png)