Update app.py

app.py

@@ -1,17 +1,33 @@
-import spaces
 import gradio as gr
+import spaces
+import torch
+from diffusers import AutoencoderKL, TCDScheduler
+from diffusers.models.model_loading_utils import load_state_dict
 from gradio_imageslider import ImageSlider
-# from PIL import Image
+from huggingface_hub import hf_hub_download
+
+from controlnet_union import ControlNetModel_Union
+from pipeline_fill_sd_xl import StableDiffusionXLFillPipeline
+
+from PIL import Image, ImageDraw
 import numpy as np
-import torch
 
 with gr.Blocks() as demo:
-    
-    with gr.Row():
-        with gr.Column(scale=1):
-            input_image = gr.Image(label="Input Image", type="numpy")
-            process_btn = gr.Button(value="generate", variant = "primary")
-        with gr.Column(scale=1):
-            output_image = gr.Image(label="Generated Image", interactive=False)
+    with gr.Column():
+        with gr.Row():
+            with gr.Column():
+                input_image = gr.Image(
+                    type="pil",
+                    label="Input Image"
+                )
+
+                with gr.Row():
+                    with gr.Column(scale=2):
+                        prompt_input = gr.Textbox(label="Prompt (Optional)")
+                    with gr.Column(scale=1):
+                        run_button = gr.Button('Generate')
+
+            with gr.Column():
+                result = gr.Image(label="Generate Image", interactive=False)
 
-demo.launch(debug=True)
+demo.queue(max_size=12).launch(share=False)