File size: 1,524 Bytes
12863e1 2cf6be0 05d3d42 2cf6be0 92ecdcd 6b0d828 2cf6be0 12863e1 262e1d3 2cf6be0 05d3d42 92ecdcd 2cf6be0 fa0ee64 262e1d3 92ecdcd f0f8ecd 262e1d3 92ecdcd ad9ba71 2cf6be0 92ecdcd 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 33 34 35 36 37 38 39 40 |
import spaces
import gradio as gr
import torch
from diffusers import DiffusionPipeline
import rembg
model_id = "stabilityai/stable-diffusion-xl-base-1.0"
pipe = DiffusionPipeline.from_pretrained(model_id,
torch_dtype=torch.float16,
use_safetensors=True,)
pipe.load_lora_weights("CiroN2022/shoes", weight_name="shoes.safetensors", adapter_name="shoes")
pipe.set_adapters("shoes")
pipe.to("cuda")
# Function to generate an image from text using diffusion
@spaces.GPU
def generate_image(prompt):
image = pipe(prompt).images
image2 = rembg.remove(image)
return image, image2
_TITLE = "Shoe Generator"
with gr.Blocks(_TITLE) as ShoeGen:
with gr.Row():
with gr.Column():
prompt = gr.Textbox(label="Enter a prompt")
neg_prompt = gr.Textbox(label="Enter a negative prompt", value="low quality, watermark, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, extra limbs, body out of frame, blurry, bad anatomy, blurred, watermark, grainy, signature, cut off, draft, closed eyes, text, logo")
button_gen = gr.Button("Generate Image")
with gr.Column():
image = gr.Image(label="Generated Image", show_download_button=True)
image2 = gr.Image(label="Generated Image without background", show_download_button=True)
button_gen.click(generate_image, inputs=[prompt], outputs=[image, image2])
ShoeGen.launch()
|