git clone https://github.com/nextgrid/streamlit_gpt3_boilerplate

source venv/bin/activate pip install -r requirements.txt pip install stability-sdk mport io import stability_sdk.interfaces.gooseai.generation.generation_pb2 as generation from PIL import Image from stability_sdk import clien

def stability_generation(self, inp, api_key): stability_api = client.StabilityInference( key=api_key, verbose=True, # Print debug messages. engine="stable-diffusion-512-v2-1", )

    # Create a request.
    result = stability_api.generate(
        prompt=inp, steps=40, samples=1, sampler=generation.SAMPLER_K_DPMPP_2M
    )

    for response in result:
        for artifact in response.artifacts:
            if artifact.type == generation.ARTIFACT_IMAGE:
                img = Image.open(io.BytesIO(artifact.binary))

                # return 1st image
                return img

Amranesara / Ss

git clone https://github.com/nextgrid/streamlit_gpt3_boilerplate #1