Spaces:
Running
Running
| import spaces | |
| import gradio as gr | |
| from PIL import Image | |
| from diffusers import StableDiffusionImg2ImgPipeline | |
| import torch | |
| model_id_or_path = os.environ.get('MODEL') | |
| pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16,safety_checker=None, use_safetensors=True) | |
| pipe = pipe.to('cuda') | |
| def generate(image_editor): | |
| init_image = image_editor['composite'].convert('RGB') | |
| #img = handle_image_to_image(init_image, prompt='photorealistic, fashion model', negative_prompt='mannequin, sketch, drawing') | |
| init_image.thumbnail((1024, 1024)) | |
| prompt = "photorealistic, fashion model" | |
| negative_prompt='mannequin, sketch, drawing, animated, anime, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, extra limbs, disfigured, deformed, body out of frame, bad anatomy, watermark, signature, cut off, low contrast, underexposed, overexposed, bad art, beginner, amateur, distorted face, deformed face, blurry, draft, grainy' | |
| pipe.to('cuda') | |
| images = pipe(prompt=prompt, image=init_image, strength=0.65, guidance_scale=7.5,negative_prompt=negative_prompt).images | |
| img = images[0] | |
| return img | |
| iface = gr.Interface(fn=generate, inputs=[gr.ImageEditor(label='Sketchpad' ,type='pil', height=700, value={'background':'./sketch1.png', 'layers':None, 'composite':None}, sources=['upload'], transforms=[]), | |
| #gr.Textbox(label="Prompt", value="photorealistic, fashion model", visible=False), | |
| #gr.Textbox(label="Negative Prompt", value='mannequin, sketch, drawing', visible=False) | |
| ], | |
| outputs=gr.Image(label="Output Image"), | |
| title="Sketch to Fashion Design", | |
| description=""" | |
| by [Tony Assi](https://www.tonyassi.com/) | |
| Please ❤️ this Space. | |
| I build custom AI apps for companies. <a href="mailto: tony.assi.media@gmail.com">Email me</a> for business inquiries. | |
| """, | |
| api_name=False, | |
| theme = gr.themes.Base(primary_hue="teal",secondary_hue="teal",neutral_hue="slate"), | |
| cache_examples=True, | |
| examples=[[{'background':'./example1.png', 'layers':None, 'composite':'./example1.png'}], [{'background':'./example2.png', 'layers':None, 'composite':'./example2.png'}],[{'background':'./sketch1.png', 'layers':None, 'composite':'./sketch1.png'}]]) | |
| iface.launch() |