Update app.py

app.py

@@ -25,7 +25,15 @@ def generate_images(prompt: str, negative_prompt:str, num_imgs_to_gen: int, num_
     return generated_img
     
 with gr.Blocks() as demo:
-    gr.HTML("<h2 style=\"font-size: 2em; font-weight: bold\" align=\"center\"> Dreambooth Galaxy Mergers </h2>")    
+    gr.Markdown(f""""
+    # Keras-Dreambooth Galaxy Mergers 🪐🤗
+
+    This is a Keras Dreambooth model fine-tuned to images of galaxy mergers taken with the [Hubble Space Telescope](https://en.wikipedia.org/wiki/Hubble_Space_Telescope) (credit [ESA/Hubble](https://esahubble.org/images/)).
+    The model, part of the [Keras Dreambooth Sprint](https://github.com/huggingface/community-events/tree/main/keras-dreambooth-sprint) trained by lposti, can be found in [keras-dreambooth/dreambooth-galaxy-mergers](https://huggingface.co/keras-dreambooth/dreambooth-galaxy-mergers).
+    
+    The model should be used with a prompt containing `sks galaxies merging`. A typical prompt for this model is `image of sks galaxies merging in space`.
+    
+    """)
     with gr.Row():
         with gr.Column():
             prompt = gr.Textbox(lines=1, value="image of sks galaxies merging in space", label="Base Prompt")
@@ -39,8 +47,20 @@ with gr.Blocks() as demo:
 
     run.click(generate_images, inputs=[prompt,negative_prompt, samples, num_steps, ugs], outputs=gallery)
     
-    gr.Examples([["image of monument in sks style, 8k, high quality, old paper","colored, deformed, blurry, grain, artifacts, low quality", 1, 30, 18],
+    gr.Examples([["image of sks galaxies merging in space, 8k, high quality","deformed, blurry, grain, artifacts, low quality", 1, 40, 12],
+                 ["painting of sks galaxies merging in van gogh style, 8k, high quality, trending on artstation","deformed, blurry, grain, artifacts, low quality", 1, 40, 12],
                  ],
                 [prompt,negative_prompt, samples,num_steps, ugs], gallery, generate_images)
 
+    gr.Markdown(f""""
+    ## Output samples
+
+    > image of sks galaxies merging in space
+    ![image of sks galaxies merging in space, 8k, high quality](https://huggingface.co/keras-dreambooth/dreambooth-galaxy-mergers/resolve/main/output-samples-mergers/db-mergers_collage.jpg)
+
+    > painting of sks galaxies merging in van gogh style
+    ![painting of sks galaxies merging in van gogh style, 8k, high quality, trending on artstation](https://huggingface.co/keras-dreambooth/dreambooth-galaxy-mergers/resolve/main/output-samples-mergers/db-mergers-vangogh_collage.jpg)
+    
+    """)
+
 demo.launch(debug=True)