| import gradio as gr | |
| from diffusers import StableDiffusionPipeline | |
| import torch | |
| pipe = StableDiffusionPipeline.from_pretrained("MohamedRashad/diffusion_fashion", torch_dtype=torch.float32) | |
| pipe.to("cpu") | |
| def generate_image(text): | |
| images = pipe(text).images | |
| image = images[0] | |
| return image | |
| diffusion_interface = gr.Interface( | |
| generate_image, | |
| gr.Textbox(lines=1, label="Input"), | |
| gr.Image(type="pil", label="Output"), | |
| title="Diffusion4Fashion: Generate cool clothes!", | |
| description="<center><p>Enter a description about a piece of cloth and the model will generate an image.</p></center>", | |
| examples=["A photo of a dress, made in 2019, color is Red, Casual usage, Women's cloth, something for the summer season, on white background"], | |
| ) | |
| diffusion_interface.launch() |