File size: 570 Bytes
9236d71
 
 
 
 
cfcf4ad
1db9185
9236d71
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
cfcf4ad
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

import requests
import os


secret_key = os.getenv("HF_TOKEN")
secret_key = "Bearer "+ secret_key

API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large"
headers = {"Authorization": secret_key}

def query(payload):
	response = requests.post(API_URL, headers=headers, json=payload)
	return response.content

# You can access the image with PIL.Image for example

import io
from PIL import Image
def getImage(text):
    image_bytes = query({
	"inputs": text,
    })
    image = Image.open(io.BytesIO(image_bytes))
    return image