ability to set sizes and cancel

app.py

@@ -27,8 +27,9 @@ DEFAULT_NEGATIVE_PROMPT = None
 @spaces.GPU
 def infer(
     prompt=None,
-    #ref=None,
     seed=0,
+    width=512,
+    height=512,
     text_cfg=7.5,
     img_cfg=1.0,
     num_inference_steps=28,
@@ -36,17 +37,12 @@ def infer(
     negative_prompt=DEFAULT_NEGATIVE_PROMPT,
     progress=gr.Progress(track_tqdm=True),
 ):
-    #if ref is None:
-    #    gr.Warning("⚠️ Please upload an image!")
-    #    return None
-
     if prompt in [None, ""]:
         gr.Warning("⚠️ Please enter a prompt!")
         return None
-    IMG_SIZE = 512
     image = pipeline.generate_image(
         editing_caption,
-        hw=(IMG_SIZE, IMG_SIZE),
+        hw=(height, width),
         num_images_per_caption=1,
         positive_prompt=positive_prompt,
         negative_prompt=negative_prompt,
@@ -85,10 +81,9 @@ with gr.Blocks(css=css) as demo:
             editing_caption = prompt
 
             run_button = gr.Button("Run", scale=0, variant="primary")
+            cancel_button = gr.Button("Cancel", scale=0, variant="secondary")
 
         with gr.Row():
-            #ref = gr.Image(label="Reference Image", show_label=True, type="pil", height=400)
-
             with gr.Accordion("Advanced Settings", open=True):
                 positive_prompt = gr.Text(
                     label="Positive Prompt",
@@ -113,45 +108,45 @@ with gr.Blocks(css=css) as demo:
                         value=3407,
                     )
                     num_inference_steps = gr.Slider(
-                        label="# sampling steps",
+                        label="Sampling steps",
                         minimum=10,
                         maximum=50,
                         step=1,
-                        value=28,  # Replace with defaults that work for your model
+                        value=28,
+                    )
+                
+                with gr.Row():
+                    width = gr.Slider(
+                        label="Width",
+                        minimum=256,
+                        maximum=1024,
+                        step=64,
+                        value=512,
+                    )
+                    height = gr.Slider(
+                        label="Height",
+                        minimum=256,
+                        maximum=1024,
+                        step=64,
+                        value=512,
                     )
-
-                #with gr.Row():
-                    #text_cfg = gr.Slider(
-                    #    label="Text cfg",
-                    #    minimum=1.0,
-                    #    maximum=15.0,
-                    #    step=0.1,
-                    #    value=7.5,  # Replace with defaults that work for your model
-                    #)
-                    #img_cfg = gr.Slider(
-                    #    label="Image cfg",
-                    #    minimum=1.0,
-                    #    maximum=15.0,
-                    #    step=0.1,
-                    #    value=1.0,
-                    #)
 
         with gr.Row():
             result_1 = gr.Image(label="Result 1", show_label=False, container=True, height=400, visible=False)
-            #result_2 = gr.Image(label="Result 2", show_label=False, container=True, height=400, visible=False)
 
         #gr.Examples(examples=examples, inputs=[prompt, ref])
 
     def show_result():
         return gr.update(visible=True)
 
-    gr.on(
+    generation_event = gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,
         inputs=[
             prompt,
-            #ref,
             seed,
+            width,
+            height,
             text_cfg,
             img_cfg,
             num_inference_steps,
@@ -160,6 +155,13 @@ with gr.Blocks(css=css) as demo:
         ],
         outputs=[result_1],
     )
+    
+    cancel_button.click(
+        fn=None,
+        inputs=None,
+        outputs=None,
+        cancels=[generation_event]
+    )
 
     gr.on(
         triggers=[run_button.click, prompt.submit],