app.py

import spaces
import gradio as gr
from PIL import Image
from diffusers import StableDiffusionImg2ImgPipeline
import torch
import os

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')

@spaces.GPU(enable_queue=True)
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: [email protected]">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()