Added sample

app.py

@@ -22,6 +22,13 @@ def main():
         with torch.no_grad():
             imgs = generate_images(dog_path if fn_is_dog else human_path, 2, 1)
         return imgs
+
+    def load_examples():
+        torch.manual_seed(32)
+        # print(fn_is_dog)
+        with torch.no_grad():
+            imgs = generate_images(dog_path, 2, 1)
+        return 32, 1, imgs[0], imgs[1]
     
     with gr.Blocks() as demo:
         gr.HTML("""
@@ -41,6 +48,7 @@ def main():
                 interactive=True
             )
             generate_button = gr.Button("Generate")
+            sample_button = gr.Button("Load Example")
 
         with gr.Row():
             g1 = gr.Gallery(
@@ -59,6 +67,12 @@ def main():
             outputs=[g1, g2]
         )
 
+        sample_button.click(
+            fn=load_examples,
+            inputs=[],
+            outputs=[seed, is_dog, 1, g2]
+        )
+
     demo.queue(concurrency_count=1)
     demo.launch(share=False, server_name="0.0.0.0")