File size: 722 Bytes
12863e1 2cf6be0 262e1d3 6b0d828 2cf6be0 12863e1 262e1d3 2cf6be0 262e1d3 2cf6be0 fa0ee64 262e1d3 f0f8ecd 262e1d3 fa0ee64 2cf6be0 262e1d3 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 |
import spaces
import gradio as gr
import torch
from diffusers import DiffusionPipeline
model_id = "CiroN2022/shoes"
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)
return images[0]
_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():
output = gr.Image(label="Generated Image")
button_gen.click(generate_image, inputs=[prompt], outputs=output)
ShoeGen.launch()
|