This is an example of an Opendream serverless extension using Baseten that overrides the default `dream` function.

baseten_stable_diffusion.py

import baseten
import base64

from opendream import opendream
from opendream.layer import ImageLayer, Layer

@opendream.define_op
def dream(prompt: str):
    # https://app.baseten.co/explore/stable_diffusion

    # You can retrieve your deployed model version ID from the UI
    # TODO: Deploy on Baseten and add your version ID
    model = baseten.deployed_model_version_id('YOUR_VERSION_ID')

    request = {
        "prompt": f"{prompt}",
        "scheduler": "ddim",
        "negative_prompt": "blurry, disfigured, low resolution"
    }

    response = model.predict(request)
    img=Layer.b64_to_pil(response["data"][0])
    
    return ImageLayer(image=img)