change seed

app.py

@@ -77,7 +77,7 @@ MAX_IMAGE_SIZE = 1024
 @spaces.GPU
 def infer_depth(prompt, 
           image = None, 
-          negative_prompt = "", 
+          negative_prompt = "nsfw，脸部阴影，低分辨率，jpeg伪影、模糊、糟糕，黑脸，霓虹灯", 
           seed = 397886929, 
           randomize_seed = False,
           guidance_scale = 6.0, 
@@ -105,12 +105,12 @@ def infer_depth(prompt,
         num_images_per_prompt=1,
         generator=generator,
     ).images[0]
-    return [condi_img, image]
+    return [condi_img, image], seed
 
 @spaces.GPU
 def infer_canny(prompt, 
           image = None, 
-          negative_prompt = "", 
+          negative_prompt = "nsfw，脸部阴影，低分辨率，jpeg伪影、模糊、糟糕，黑脸，霓虹灯", 
           seed = 397886929, 
           randomize_seed = False,
           guidance_scale = 6.0, 
@@ -138,7 +138,7 @@ def infer_canny(prompt,
         num_images_per_prompt=1,
         generator=generator,
     ).images[0]
-    return [condi_img, image]
+    return [condi_img, image], seed
 
 canny_examples = [
     ["一个漂亮的女孩，高品质，超清晰，色彩鲜艳，超高分辨率，最佳品质，8k，高清，4K",
@@ -244,6 +244,7 @@ with gr.Blocks(css=css) as Kolors:
             
         with gr.Column(elem_id="col-right"):
             result = gr.Gallery(label="Result", show_label=False, columns=2)
+            seed_used = gr.Number(label="Seed Used")
     
     with gr.Row():
         gr.Examples(
@@ -265,13 +266,13 @@ with gr.Blocks(css=css) as Kolors:
     canny_button.click(
         fn = infer_canny,
         inputs = [prompt, image, negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, controlnet_conditioning_scale, control_guidance_end, strength],
-        outputs = [result]
+        outputs = [result, seed_used]
     )
 
     depth_button.click(
         fn = infer_depth,
         inputs = [prompt, image, negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, controlnet_conditioning_scale, control_guidance_end, strength],
-        outputs = [result]
+        outputs = [result, seed_used]
     )
 
 Kolors.queue().launch(debug=True)