Spaces:
Runtime error
Runtime error
import gradio as gr | |
from generate import generate | |
def generate_wrapper(prompt, seed=0, negative_prompt="", model="Deliberate", sampler="k_dpmpp_2s_a", steps=50): | |
url = generate(prompt, seed=seed, negative_prompt=negative_prompt, model=model, sampler=sampler, steps=steps) | |
return {"image": url} | |
inputs = [ | |
gr.inputs.Textbox(label="Prompt"), | |
gr.inputs.Number(label="Seed", default=0), | |
gr.inputs.Textbox(label="Negative Prompt", default=""), | |
gr.inputs.Dropdown(["Deliberate", "Curie", "BigGAN"], label="Model", default="Deliberate"), | |
gr.inputs.Dropdown(["k_dpmpp_2s_a", "k_heun", "k_dpps_2s_a"], label="Sampler", default="k_dpmpp_2s_a"), | |
gr.inputs.Number(label="Steps", default=50) | |
] | |
outputs = gr.outputs.Image(type="pil") | |
interface = gr.Interface(generate_wrapper, inputs, outputs, title="Generate Images from Prompts", | |
description="Enter a prompt and generate an image using this API.", | |
examples=[["A cat sitting on a chair"], | |
["A landscape with mountains and a lake"], | |
["A portrait of a woman with red hair"]]) | |
interface.launch() | |