update sdxl

app.py

@@ -12,6 +12,7 @@ transformers.utils.move_cache()
 from diffusers import StableDiffusionXLControlNetPipeline, ControlNetModel
 from transformers import BlipProcessor, BlipForConditionalGeneration
 from accelerate import Accelerator
+torch.set_num_threads(2)
 
 import warnings
 warnings.filterwarnings("ignore") 
@@ -147,7 +148,7 @@ def colorize_image(input_image, mode):
 
 def gradio_interface():
     with gr.Blocks() as app:
-        with gr.Tab("Mode Colorization no Prompting"):
+        with gr.Tab("Prompt-Free"):
             with gr.Blocks():
                 input_image = gr.Image(type="numpy", label="Upload an Image")
                 output_modes = gr.Radio(
@@ -185,24 +186,29 @@ def gradio_interface():
                     outputs=[resnet_output, mobilenet_output, autoencoder_output]
                 )
 
-        with gr.Tab("Stable Diffusion"):
+        with gr.Tab("Prompt_Guided(ControlNet-SDXL)"):
             with gr.Blocks():
                 sd_image = gr.Image(label="Upload a Color Image")
                 positive_prompt = gr.Textbox(label="Positive Prompt", placeholder="Text for positive prompt")
                 negative_prompt = gr.Textbox(
-                    value="low quality, bad quality, low contrast, black and white, bw, monochrome, grainy, blurry, historical, restored, desaturate",
-                    label="Negative Prompt", placeholder="Text for negative prompt"
+                value="low quality, bad quality, low contrast, black and white, bw, monochrome, grainy, blurry, historical, restored, desaturate",
+                label="Negative Prompt", placeholder="Text for negative prompt"
                 )
-                generate_caption = gr.Checkbox(label="Generate Caption?", value=False)
+                generate_caption = gr.Checkbox(label="Generate Caption?", value=True)
+                seed = gr.Number(label="Seed", value=123, precision=0)
+                inference_steps = gr.Number(label="Inference Steps", value=5, precision=0)
+ 
                 submit_sd = gr.Button("Generate")
 
+      
                 sd_output_image = gr.Image(label="Colorized Image")
-                sd_caption = gr.Textbox(label="Captioning Result", show_copy_button=True, visible=False)
+                sd_caption = gr.Textbox(label="Captioning Result", show_copy_button=True, visible=True)
 
+    
                 submit_sd.click(
-                    fn=colorize_image_sdxl,
-                    inputs=[sd_image, positive_prompt, negative_prompt, generate_caption],
-                    outputs=[sd_output_image, sd_caption]
+                fn=colorize_image_sdxl,
+                inputs=[sd_image, positive_prompt, negative_prompt, generate_caption, seed, inference_steps],
+                outputs=[sd_output_image, sd_caption]
                 )
 
     return app