| from diffusers import StableDiffusionPipeline as SD | |
| import torch | |
| import gradio as gr | |
| model_id = "runwayml/stable-diffusion-v1-5" | |
| gen = SD.from_pretrained(model_id) | |
| gen.to("cuda") | |
| def generate(): | |
| prompt = "A doggo" | |
| image = gen(prompt).images[0] | |
| image.save("result.png") | |
| ui = gr.Interface(fn=generate, inputs="prompt_field", outputs="result") | |
| ui.launch() |