Make width and height customizable

app.py

@@ -73,8 +73,10 @@ def inference(
     seed: int = -1,
     init_image: Image.Image | None = None,
     qrcode_image: Image.Image | None = None,
-    use_qr_code_as_init_image = True,
-    sampler = "DPM++ Karras SDE",
+    use_qr_code_as_init_image=True,
+    sampler="DPM++ Karras SDE",
+    width: int = 768,
+    height: int = 768,
 ):
     if prompt is None or prompt == "":
         raise gr.Error("Prompt is required")
@@ -111,8 +113,8 @@ def inference(
         negative_prompt=negative_prompt,
         image=qrcode_image,
         control_image=qrcode_image,  # type: ignore
-        width=768,  # type: ignore
-        height=768,  # type: ignore
+        width=width,  # dynamic width
+        height=height,  # dynamic height
         guidance_scale=float(guidance_scale),
         controlnet_conditioning_scale=float(controlnet_conditioning_scale),  # type: ignore
         generator=generator,
@@ -166,6 +168,7 @@ model: https://huggingface.co/DionTimmer/controlnet_qrcode-control_v1p_sd15
             )
             use_qr_code_as_init_image = gr.Checkbox(label="Use QR code as init image", value=True, interactive=False, info="Whether init image should be QR code. Unclick to pass init image or generate init image with Stable Diffusion 2.1")
 
+            
             with gr.Accordion(label="Init Images (Optional)", open=False, visible=False) as init_image_acc:
                 init_image = gr.Image(label="Init Image (Optional). Leave blank to generate image with SD 2.1", type="pil")
 
@@ -174,6 +177,21 @@ model: https://huggingface.co/DionTimmer/controlnet_qrcode-control_v1p_sd15
                 label="Params: The generated QR Code functionality is largely influenced by the parameters detailed below",
                 open=True,
             ):
+                width = gr.Slider(
+                    minimum=64,
+                    maximum=2048,
+                    step=64,
+                    value=768,
+                    label="Width",
+                )
+                height = gr.Slider(
+                    minimum=64,
+                    maximum=2048,
+                    step=64,
+                    value=768,
+                    label="Height",
+                )
+                
                 controlnet_conditioning_scale = gr.Slider(
                     minimum=0.0,
                     maximum=5.0,
@@ -218,6 +236,8 @@ model: https://huggingface.co/DionTimmer/controlnet_qrcode-control_v1p_sd15
             qr_code_image,
             use_qr_code_as_init_image,
             sampler,
+            width,  # add width input
+            height,  # add height input
         ],
         outputs=[result_image],
         concurrency_limit=1