import gradio as gr import numpy as np import os import time import math import random import imageio import torch import spaces from diffusers import ( ControlNetModel, StableDiffusionControlNetPipeline, ) from PIL import Image max_64_bit_int = np.iinfo(np.int32).max if torch.cuda.is_available(): device = "cuda" floatType = torch.float16 else: device = "cpu" floatType = torch.float32 controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11e_sd15_ip2p", torch_dtype = floatType) pipe = StableDiffusionControlNetPipeline.from_pretrained( "runwayml/stable-diffusion-v1-5", safety_checker = None, controlnet = controlnet, torch_dtype = floatType ) pipe = pipe.to(device) def update_seed(is_randomize_seed, seed): if is_randomize_seed: return random.randint(0, max_64_bit_int) return seed def check( input_image, prompt, negative_prompt, denoising_steps, num_inference_steps, guidance_scale, image_guidance_scale, is_randomize_seed, seed, progress = gr.Progress()): if input_image is None: raise gr.Error("Please provide an image.") if prompt is None or prompt == "": raise gr.Error("Please provide a prompt input.") @spaces.GPU(duration=420) def pix2pix( input_image, prompt, negative_prompt, denoising_steps, num_inference_steps, guidance_scale, image_guidance_scale, is_randomize_seed, seed, progress = gr.Progress()): check( input_image, prompt, negative_prompt, denoising_steps, num_inference_steps, guidance_scale, image_guidance_scale, is_randomize_seed, seed ) start = time.time() progress(0, desc = "Preparing data...") if negative_prompt is None: negative_prompt = "" if denoising_steps is None: denoising_steps = 0 if num_inference_steps is None: num_inference_steps = 20 if guidance_scale is None: guidance_scale = 5 if image_guidance_scale is None: image_guidance_scale = 1.5 if seed is None: seed = random.randint(0, max_64_bit_int) random.seed(seed) torch.manual_seed(seed) original_height, original_width, dummy_channel = np.array(input_image).shape output_width = original_width output_height = original_height mask_image = Image.new(mode = input_image.mode, size = (output_width, output_height), color = "white") limitation = ""; # Limited to 1 million pixels if 1024 * 1024 < output_width * output_height: factor = ((1024 * 1024) / (output_width * output_height))**0.5 output_width = math.floor(output_width * factor) output_height = math.floor(output_height * factor) limitation = " Due to technical limitation, the image have been downscaled and then upscaled."; # Width and height must be multiple of 8 output_width = output_width - (output_width % 8) output_height = output_height - (output_height % 8) progress(None, desc = "Processing...") output_image = pipe( seeds=[seed], width = output_width, height = output_height, prompt = prompt, negative_prompt = negative_prompt, image = input_image, mask_image = mask_image, num_inference_steps = num_inference_steps, guidance_scale = guidance_scale, image_guidance_scale = image_guidance_scale, denoising_steps = denoising_steps, show_progress_bar = True ).images[0] if limitation != "": output_image = output_image.resize((original_width, original_height)) end = time.time() secondes = int(end - start) minutes = math.floor(secondes / 60) secondes = secondes - (minutes * 60) hours = math.floor(minutes / 60) minutes = minutes - (hours * 60) return [ output_image, ("Start again to get a different result. " if is_randomize_seed else "") + "The image has been generated in " + ((str(hours) + " h, ") if hours != 0 else "") + ((str(minutes) + " min, ") if hours != 0 or minutes != 0 else "") + str(secondes) + " sec." + limitation ] with gr.Blocks() as interface: gr.HTML( """

Instruct Pix2Pix demo

Modifies your image using a textual instruction, freely, without account, without watermark, without installation, which can be downloaded



✨ Powered by SD 1.5 and ControlNet. The result quality extremely varies depending on what we ask.

""" + ("🏃‍♀️ Estimated time: few minutes." if torch.cuda.is_available() else "🐌 Slow process... ~1 hour.") + """ Your computer must not enter into standby mode. You can launch several generations in different browser tabs when you're gone. If this space does not work or you want a faster run, use Instruct Pix2Pix available on hysts's ControlNet-v1-1 space (last tab) or on Dezgo site.
You can duplicate this space on a free account, it's designed to work on CPU, GPU and ZeroGPU.

⚖️ You can use, modify and share the generated images but not for commercial uses. """ ) with gr.Column(): input_image = gr.Image(label = "Your image", sources = ["upload", "webcam", "clipboard"], type = "pil") prompt = gr.Textbox(label = "Prompt", info = "Instruct what to change in the image", placeholder = "Order the AI what to change in the image", lines = 2) with gr.Accordion("Advanced options", open = False): negative_prompt = gr.Textbox(label = "Negative prompt", placeholder = "Describe what you do NOT want to see in the image", value = "Watermark") denoising_steps = gr.Slider(minimum = 0, maximum = 1000, value = 0, step = 1, label = "Denoising", info = "lower=irrelevant result, higher=relevant result") num_inference_steps = gr.Slider(minimum = 10, maximum = 500, value = 20, step = 1, label = "Number of inference steps", info = "lower=faster, higher=image quality") guidance_scale = gr.Slider(minimum = 1, maximum = 13, value = 5, step = 0.1, label = "Classifier-Free Guidance Scale", info = "lower=image quality, higher=follow the prompt") image_guidance_scale = gr.Slider(minimum = 1, value = 1.5, step = 0.1, label = "Image Guidance Scale", info = "lower=image quality, higher=follow the image") randomize_seed = gr.Checkbox(label = "\U0001F3B2 Randomize seed", value = True, info = "If checked, result is always different") seed = gr.Slider(minimum = 0, maximum = max_64_bit_int, step = 1, randomize = True, label = "Seed") submit = gr.Button("🚀 Modify", variant = "primary") modified_image = gr.Image(label = "Modified image") information = gr.HTML() submit.click(fn = update_seed, inputs = [ randomize_seed, seed ], outputs = [ seed ], queue = False, show_progress = False).then(check, inputs = [ input_image, prompt, negative_prompt, denoising_steps, num_inference_steps, guidance_scale, image_guidance_scale, randomize_seed, seed ], outputs = [], queue = False, show_progress = False).success(pix2pix, inputs = [ input_image, prompt, negative_prompt, denoising_steps, num_inference_steps, guidance_scale, image_guidance_scale, randomize_seed, seed ], outputs = [ modified_image, information ], scroll_to_output = True) gr.Examples( run_on_click = True, fn = pix2pix, inputs = [ input_image, prompt, negative_prompt, denoising_steps, num_inference_steps, guidance_scale, image_guidance_scale, randomize_seed, seed ], outputs = [ modified_image, information ], examples = [ [ "./Examples/Example1.webp", "What if it's snowing?", "Watermark", 1, 20, 5, 1.5, False, 42 ], [ "./Examples/Example2.png", "What if this woman had brown hair?", "Watermark", 1, 20, 5, 1.5, False, 42 ], [ "./Examples/Example3.jpeg", "Replace the house by a windmill", "Watermark", 1, 20, 5, 1.5, False, 42 ], [ "./Examples/Example4.gif", "What if the camera was in opposite side?", "Watermark", 1, 20, 5, 1.5, False, 42 ], [ "./Examples/Example5.bmp", "Turn him into cyborg", "Watermark", 1, 20, 5, 25, False, 42 ], ], cache_examples = False, ) interface.queue().launch()