Update seed

app.py

@@ -1,8 +1,3 @@
-from diffusers import (
-    ControlNetModel,
-    DiffusionPipeline,
-    StableDiffusionControlNetPipeline,
-)
 import gradio as gr
 import numpy as np
 import os
@@ -10,19 +5,31 @@ import time
 import math
 import random
 import imageio
-from PIL import Image, ImageFilter
 import torch
 
+from diffusers import (
+    ControlNetModel,
+    DiffusionPipeline,
+    StableDiffusionControlNetPipeline,
+)
+from PIL import Image, ImageFilter
+
 max_64_bit_int = 2**63 - 1
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 floatType = torch.float16 if torch.cuda.is_available() else 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,
@@ -31,7 +38,6 @@ def check(
     num_inference_steps,
     guidance_scale,
     image_guidance_scale,
-    randomize_seed,
     seed,
     progress = gr.Progress()):
     if input_image is None:
@@ -48,7 +54,6 @@ def pix2pix(
     num_inference_steps,
     guidance_scale,
     image_guidance_scale,
-    randomize_seed,
     seed,
     progress = gr.Progress()):
     check(
@@ -59,7 +64,6 @@ def pix2pix(
         num_inference_steps,
         guidance_scale,
         image_guidance_scale,
-        randomize_seed,
         seed
     )
     start = time.time()
@@ -80,11 +84,11 @@ def pix2pix(
     if image_guidance_scale is None:
         image_guidance_scale = 1.5
 
-    if randomize_seed:
+    if seed is None:
         seed = random.randint(0, max_64_bit_int)
 
     random.seed(seed)
-    #pipe = pipe.manual_seed(seed)
+    torch.manual_seed(seed)
 
     original_height, original_width, dummy_channel = np.array(input_image).shape
     output_width = original_width
@@ -161,7 +165,7 @@ with gr.Blocks() as interface:
         """
     )
     with gr.Column():
-        input_image = gr.Image(label = "Your image", sources = ["upload"], type = "pil")
+        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')
         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')
@@ -169,15 +173,20 @@ with gr.Blocks() as interface:
             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 (not working, always checked)", 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 (if not randomized)")
+            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.Label(label = "Information")
 
-    submit.click(check, inputs = [
+    submit.click(fn = update_seed, inputs = [
+        randomize_seed,
+        seed
+    ], outputs = [
+        seed
+    ], queue = False, show_progress = False).then(check, inputs = [
         input_image,
         prompt,
         negative_prompt,
@@ -185,7 +194,6 @@ with gr.Blocks() as interface:
         num_inference_steps,
         guidance_scale,
         image_guidance_scale,
-        randomize_seed,
         seed
     ], outputs = [], queue = False, show_progress = False).success(pix2pix, inputs = [
         input_image,
@@ -195,7 +203,6 @@ with gr.Blocks() as interface:
         num_inference_steps,
         guidance_scale,
         image_guidance_scale,
-        randomize_seed,
         seed
     ], outputs = [
         modified_image,
@@ -203,6 +210,7 @@ with gr.Blocks() as interface:
     ], scroll_to_output = True)
 
     gr.Examples(
+        fn = pix2pix,
 	    inputs = [
             input_image,
             prompt,
@@ -211,7 +219,6 @@ with gr.Blocks() as interface:
             num_inference_steps,
             guidance_scale,
             image_guidance_scale,
-            randomize_seed,
             seed
         ],
 	    outputs = [
@@ -220,58 +227,53 @@ with gr.Blocks() as interface:
         ],
         examples = [
                 [
-                    "Example1.webp",
+                    "./Examples/Example1.webp",
                     "What if it's snowing?",
                     "Watermark",
                     1,
                     20,
                     5,
                     1.5,
-                    True,
                     42
                 ],
                 [
-                    "Example2.png",
+                    "./Examples/Example2.png",
                     "What if this woman had brown hair?",
                     "Watermark",
                     1,
                     20,
                     5,
                     1.5,
-                    True,
                     42
                 ],
                 [
-                    "Example3.jpeg",
+                    "./Examples/Example3.jpeg",
                     "Replace the house by a windmill",
                     "Watermark",
                     1,
                     20,
                     5,
                     1.5,
-                    True,
                     42
                 ],
                 [
-                    "Example4.gif",
+                    "./Examples/Example4.gif",
                     "What if the camera was in opposite side?",
                     "Watermark",
                     1,
                     20,
                     5,
                     1.5,
-                    True,
                     42
                 ],
                 [
-                    "Example5.bmp",
+                    "./Examples/Example5.bmp",
                     "Turn him into cyborg",
                     "Watermark",
                     1,
                     20,
                     5,
                     25,
-                    True,
                     42
                 ],
             ],