| import io |
| import os |
| import warnings |
|
|
| from PIL import Image |
| from stability_sdk import client |
| import stability_sdk.interfaces.gooseai.generation.generation_pb2 as generation |
|
|
| import gradio as gr |
| stability_api = client.StabilityInference( |
| key=os.environ["Secret"], |
| verbose=True, |
| ) |
|
|
|
|
| def infer(prompt): |
| |
| answers = stability_api.generate( |
| prompt=f"Beautiful Portait of a {prompt} made out of flowers 💐 🌺 🌸 , artstation winner by Victo Ngai, Kilian Eng, vibrant colors, winning-award masterpiece, aesthetic octane render, 8K HD", |
| height =640 |
| ) |
| |
| |
| for resp in answers: |
| for artifact in resp.artifacts: |
| if artifact.finish_reason == generation.FILTER: |
| warnings.warn( |
| "Your request activated the API's safety filters and could not be processed." |
| "Please modify the prompt and try again.") |
| if artifact.type == generation.ARTIFACT_IMAGE: |
| img = Image.open(io.BytesIO(artifact.binary)) |
| return img |
|
|
|
|
| block = gr.Blocks(css=".container { max-width: 600px; margin: auto; }") |
|
|
| num_samples = 1 |
|
|
|
|
|
|
| with block as demo: |
| gr.Markdown("<h1><center>Flower Diffusion</center></h1>") |
| gr.Markdown( |
| "Get a pretty flowery image from any prompt - keep it simple!" |
| ) |
| with gr.Group(): |
| with gr.Box(): |
| with gr.Row().style(mobile_collapse=False, equal_height=True): |
|
|
| text = gr.Textbox( |
| value = "Kitty cat", |
| label="Enter your prompt", show_label=False, max_lines=1 |
| ).style( |
| border=(True, False, True, True), |
| rounded=(True, False, False, True), |
| container=False, |
| ) |
| btn = gr.Button("Run").style( |
| margin=False, |
| rounded=(False, True, True, False), |
| ) |
| |
| |
| gallery = gr.Image() |
| text.submit(infer, inputs=[text], outputs=gallery) |
| btn.click(infer, inputs=[text], outputs=gallery) |
|
|
| |
| |
|
|
|
|
| demo.launch(debug=True, enable_queue = True) |
