File size: 844 Bytes
12863e1 2cf6be0 2457f9c 6b0d828 2cf6be0 12863e1 262e1d3 2cf6be0 7d79b00 2cf6be0 fa0ee64 262e1d3 f0f8ecd 262e1d3 7d79b00 ad9ba71 2cf6be0 ad9ba71 2cf6be0 fa0ee64 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import spaces
import gradio as gr
import torch
from diffusers import DiffusionPipeline
model_id = "stabilityai/stable-diffusion-2-1"
pipe = DiffusionPipeline.from_pretrained(model_id)
pipe.to("cuda")
# Function to generate an image from text using diffusion
@spaces.GPU
def generate_image(prompt):
images = pipe(prompt).images
return images
_TITLE = "Shoe Generator"
with gr.Blocks(_TITLE) as ShoeGen:
with gr.Row():
with gr.Column():
prompt = gr.Textbox(label="Enter a prompt")
button_gen = gr.Button("Generate Image")
with gr.Column():
# show images
gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
button_gen.click(generate_image, inputs=[prompt], outputs=gallery)
ShoeGen.launch()
|