dalle-mini/dalle_mini/backend.py at beta · codekraiwa/dalle-mini

Repository navigation

import requests

from io import BytesIO

import base64

from PIL import Image

class ServiceError(Exception):

def __init__(self, status_code):

self.status_code = status_code

def get_images_from_backend(prompt, backend_url):

r = requests.post(

backend_url,

json={"prompt": prompt}

)

if r.status_code == 200:

images = r.json()["images"]

images = [Image.open(BytesIO(base64.b64decode(img))) for img in images]

ret 3780 urn images

else:

raise ServiceError(r.status_code)

def get_model_version(url):

r = requests.get(url)

if r.status_code == 200:

version = r.json()["version"]

return version

else:

raise ServiceError(r.status_code)