Added more examples, moved some things

app.py

@@ -23,12 +23,21 @@ p_params = replicate(params)
 
 # Description
 title = "# 🧨 ControlNet on Segment Anything 🤗"
-description = "This is a demo on ControlNet based on Segment Anything"
+description = """This is a demo on 🧨 ControlNet based on Meta's [Segment Anything Model](https://segment-anything.com/).
+
+                Upload a Segment Anything Segmentation Map, write a prompt, and generate images 🤗 This demo is still Work in Progress, so don't expect it to work well for now !! 
+
+                
+                Test some of the examples below to give it a try ⬇️
+              """
+
+examples = [["a modern main room of a house", "low quality", "examples/condition_image_1.png"],
+            ["new york buildings,  Vincent Van Gogh starry night ", "low quality, monochrome", "examples/condition_image_2.png"],
+            ["contemporary living room,  high quality, 4k, realistic", "low quality, monochrome, low res", "examples/condition_image_3.png"]]
 
-examples = [["a modern main room of a house", "low quality", "condition_image_1.png", 50, 4, 4]]
 
 # Inference Function
-def infer(prompts, negative_prompts, image, num_inference_steps, seed, num_samples):
+def infer(prompts, negative_prompts, image, num_inference_steps = 50, seed = 4, num_samples = 4):
     rng = jax.random.PRNGKey(int(seed))
     num_inference_steps = int(num_inference_steps)
     image = Image.fromarray(image, mode="RGB")
@@ -71,10 +80,12 @@ with gr.Blocks(css="h1 { text-align: center }") as demo:
 
     # Images
     with gr.Row(variant="panel"):
-        cond_img = gr.Image(label="Input")\
-                     .style(height=400)
-        output = gr.Gallery(label="Generated images")\
-                   .style(height="auto", rows=[2], columns=[1, 2])
+        with gr.Column(scale=2):
+            cond_img = gr.Image(label="Input")\
+                    .style(height=200)
+        with gr.Column(scale=1):
+            output = gr.Gallery(label="Generated images")\
+                    .style(height=200, rows=[2], columns=[1, 2], object_fit="contain")
         
     # Submit & Clear
     with gr.Row():
@@ -85,14 +96,15 @@ with gr.Blocks(css="h1 { text-align: center }") as demo:
         with gr.Column():
             with gr.Accordion("Advanced options", open=False):
                 num_steps = gr.Slider(10, 60, 50, step=1, label="Steps")
-                seed = gr.Slider(0, 1024, 0, step=1, label="Seed")
+                seed = gr.Slider(0, 1024, 4, step=1, label="Seed")
                 num_samples = gr.Slider(1, 4, 4, step=1, label="Nº Samples")
                 
-            submit = gr.Button("Submit")
+            submit = gr.Button("Generate")
+            # TODO: Download Button
 
     # Examples
     gr.Examples(examples=examples,
-                inputs=[prompt, negative_prompt, cond_img, num_steps, seed, num_samples],
+                inputs=[prompt, negative_prompt, cond_img],
                 outputs=output,
                 fn=infer,
                 cache_examples=True)