Add load example button

app.py

@@ -16,6 +16,29 @@ help_text = """"""
 def main():
     segmodel, model, diffusion, ldm, bert, clip_model, model_params = make_models()
 
+    def load_sample():
+        SAMPLE_IMAGE = "./sample/flower1.jpg"
+        input_image = Image.open(SAMPLE_IMAGE)
+        from_text = "a flower"
+        instruction = "a sunflower"
+        negative_prompt = ""
+        seed = 42
+        guidance_scale = 5.0
+        clip_guidance_scale = 150
+        cutn = 16
+        l2_sim_lambda = 10_000
+
+        edited_image_1 = run_model(
+            segmodel, model, diffusion, ldm, bert, clip_model, model_params,
+            from_text, instruction, negative_prompt, input_image.convert('RGB'), seed, guidance_scale, clip_guidance_scale, cutn, l2_sim_lambda
+        )
+        
+        return [
+            input_image, from_text, instruction, negative_prompt, seed, guidance_scale, 
+            clip_guidance_scale, cutn, l2_sim_lambda, edited_image_1
+        ]
+
+
     def generate(
         input_image: Image.Image,
         from_text: str,
@@ -44,7 +67,7 @@ def main():
 
     def reset():
         return [
-            "Randomize Seed", 1371, None, 5.0,
+            "Randomize Seed", 42, None, 5.0,
             150, 16, 10000
         ]
 
@@ -57,8 +80,8 @@ def main():
         with gr.Row():
             with gr.Column(scale=1, min_width=100):
                 generate_button = gr.Button("Generate")
-            # with gr.Column(scale=1, min_width=100):
-            #     load_button = gr.Button("Load Example")
+            with gr.Column(scale=1, min_width=100):
+                load_button = gr.Button("Load Example")
             with gr.Column(scale=1, min_width=100):
                 reset_button = gr.Button("Reset")
             with gr.Column(scale=3):
@@ -96,20 +119,17 @@ def main():
 
         generate_button.click(
             fn=generate,
-            inputs=[
-                input_image,
-                from_text,
-                instruction,
-                negative_prompt,
-                randomize_seed,
-                seed,
-                guidance_scale,
-                clip_guidance_scale,
-                cutn,
-                l2_sim_lambda
-            ],
+            inputs=[],
             outputs=[seed, edited_image_1],
         )
+
+        load_button.click(
+            fn=load_sample,
+            inputs=[],
+            outputs=[input_image, from_text, instruction, negative_prompt, seed, guidance_scale, clip_guidance_scale, cutn, l2_sim_lambda, edited_image_1],
+        )
+
+
         reset_button.click(
             fn=reset,
             inputs=[],