update

app.py

@@ -50,8 +50,8 @@ def run_normal(prompt, negative_prompt="", guidance_scale=7, progress=gr.Progres
     return pipe_normal(prompt, negative_prompt=negative_prompt, guidance_scale=guidance_scale, num_inference_steps=20).images[0]
 
 #@spaces.GPU
-def run_edit(image, prompt, negative_prompt="", guidance_scale=7, progress=gr.Progress(track_tqdm=True)):
-    resolution = 1024
+def run_edit(image, prompt, resolution, negative_prompt="", guidance_scale=7, progress=gr.Progress(track_tqdm=True)):
+    #resolution = 1024
     image.resize((resolution, resolution))
     return pipe_edit(prompt=prompt,image=image,height=resolution,width=resolution,negative_prompt=negative_prompt, guidance_scale=guidance_scale,num_inference_steps=20).images[0]
 css = '''
@@ -70,13 +70,14 @@ with gr.Blocks(css=css) as demo:
       with gr.Group():
           image_edit = gr.Image(label="Image you would like to edit", type="pil")
           with gr.Row():
-            prompt_edit = gr.Textbox(show_label=False, scale=4, placeholder="Edit instructions, e.g.: Make the day cloudy")
+            prompt_edit = gr.Textbox(label="Prompt", scale=4, placeholder="Edit instructions, e.g.: Make the day cloudy")
+            size_edit = gr.Number(label="Size", value=1024, maximum=1024, minimum=512, precision=0)
             button_edit = gr.Button("Generate", min_width=120)
           output_edit = gr.Image(label="Your result image", interactive=False)
           with gr.Accordion("Advanced Settings", open=False):
             negative_prompt_edit = gr.Textbox(label="Negative Prompt")
             guidance_scale_edit = gr.Number(label="Guidance Scale", value=7)
-      gr.Examples(examples=edit_examples, fn=run_edit, inputs=[image_edit, prompt_edit], outputs=[output_edit], cache_examples=False)
+      gr.Examples(examples=edit_examples, fn=run_edit, inputs=[image_edit, prompt_edit, size_edit], outputs=[output_edit], cache_examples=False)
     with gr.Tab("CosXL"):
       with gr.Group():
           with gr.Row():
@@ -105,7 +106,7 @@ with gr.Blocks(css=css) as demo:
             prompt_edit.submit
         ],
         fn=run_edit,
-        inputs=[image_edit, prompt_edit, negative_prompt_edit, guidance_scale_edit],
+        inputs=[image_edit, prompt_edit, size_edit, negative_prompt_edit, guidance_scale_edit],
         outputs=[output_edit]
     )
 if __name__ == "__main__":