From f8e9c26b1abb7464f92688f533989d01769900ea Mon Sep 17 00:00:00 2001
From: Akshat Bubna <akshatb42@gmail.com>
Date: Tue, 27 Feb 2024 20:27:30 -0500
Subject: [PATCH] Update default prompt for SDXL examples (#605)

---
 06_gpu_and_ml/stable_diffusion/frontend/index.html              | 2 +-
 06_gpu_and_ml/stable_diffusion/playground.py                    | 1 +
 06_gpu_and_ml/stable_diffusion/stable_diffusion_xl.py           | 1 +
 06_gpu_and_ml/stable_diffusion/stable_diffusion_xl_lightning.py | 1 +
 4 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/06_gpu_and_ml/stable_diffusion/frontend/index.html b/06_gpu_and_ml/stable_diffusion/frontend/index.html
index db460226b..4be51174b 100644
--- a/06_gpu_and_ml/stable_diffusion/frontend/index.html
+++ b/06_gpu_and_ml/stable_diffusion/frontend/index.html
@@ -66,7 +66,7 @@ <h2 class="text-3xl font-medium text-center mb-10">
     <script>
       function state() {
         return {
-          prompt: "a beautiful Japanese temple, butterflies flying around",
+          prompt: "{{ default_prompt }}",
           submitted: "",
           loading: false,
           imageURL: "",
diff --git a/06_gpu_and_ml/stable_diffusion/playground.py b/06_gpu_and_ml/stable_diffusion/playground.py
index 66242e831..352650fdf 100644
--- a/06_gpu_and_ml/stable_diffusion/playground.py
+++ b/06_gpu_and_ml/stable_diffusion/playground.py
@@ -81,6 +81,7 @@ def app():
         html = template.render(
             inference_url=Model.inference.web_url,
             model_name="Playground 2.5",
+            default_prompt="Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
         )
         f.write(html)
 
diff --git a/06_gpu_and_ml/stable_diffusion/stable_diffusion_xl.py b/06_gpu_and_ml/stable_diffusion/stable_diffusion_xl.py
index 4af761036..b22ff49e9 100644
--- a/06_gpu_and_ml/stable_diffusion/stable_diffusion_xl.py
+++ b/06_gpu_and_ml/stable_diffusion/stable_diffusion_xl.py
@@ -182,6 +182,7 @@ def app():
         html = template.render(
             inference_url=Model.inference.web_url,
             model_name="Stable Diffusion XL",
+            default_prompt="A cinematic shot of a baby racoon wearing an intricate italian priest robe.",
         )
         f.write(html)
 
diff --git a/06_gpu_and_ml/stable_diffusion/stable_diffusion_xl_lightning.py b/06_gpu_and_ml/stable_diffusion/stable_diffusion_xl_lightning.py
index 0df3651ae..910cb61ea 100644
--- a/06_gpu_and_ml/stable_diffusion/stable_diffusion_xl_lightning.py
+++ b/06_gpu_and_ml/stable_diffusion/stable_diffusion_xl_lightning.py
@@ -88,6 +88,7 @@ def app():
         html = template.render(
             inference_url=Model.inference.web_url,
             model_name="SDXL Lightning",
+            default_prompt="A cinematic shot of a baby racoon wearing an intricate italian priest robe.",
         )
         f.write(html)