update: app + readme

README.md

@@ -1,6 +1,6 @@
 ---
-title: Sdxl Prompt Crafting
-emoji: ⚡
+title: Reproducible Stable Diffusion XL
+emoji: 🐝
 colorFrom: blue
 colorTo: blue
 sdk: gradio
@@ -10,3 +10,5 @@ pinned: false
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+This space enables us to generate images using Stable Diffusion XL and track your experiments using Weights & Biases. In order to learn more able engineering better prompts for Stable Diffusion models and how to make your experiments reproducible using [Weights & Biases](https://wandb.ai/site), check out [this report](https://wandb.ai/geekyrakshit/diffusers-prompt-engineering/reports/A-Guide-to-Prompt-Engineering-for-Diffusion-Models--Vmlldzo1NzY4NzQ3). This app is powered by [🤗 Diffusers](https://huggingface.co/docs/diffusers) and [Weights & Biases](https://wandb.ai/site). A Weights & Biases API key is necessary to run this app, which you can find at **[https://wandb.ai/authorize](https://wandb.ai/authorize)**.

app.py

@@ -1,10 +1,12 @@
 import os
+import shutil
 
 import gradio as gr
 
 import torch
 from diffusers import DiffusionPipeline
 
+import wandb
 from wandb_addons.diffusers import get_wandb_callback
 
 
@@ -35,7 +37,10 @@ def generate_image(
     seed,
     guidance_rescale,
 ):
-    os.environ["WANDB_API_KEY"] = wandb_api_key
+    if wandb_api_key is None or wandb_api_key == "":
+        raise gr.Error("A WandB API key is required to run this app! You can get one from https://wandb.ai/authorize")
+    else:
+        os.environ["WANDB_API_KEY"] = wandb_api_key
     generator = torch.Generator(device="cuda").manual_seed(seed)
     configs = {
         "guidance_scale": guidance_scale,
@@ -50,6 +55,7 @@ def generate_image(
         num_inference_steps=num_inference_steps,
         configs=configs,
     )
+    run_url = wandb.run.get_url()
     image = pipeline(
         prompt,
         negative_prompt=negative_prompt,
@@ -62,14 +68,15 @@ def generate_image(
     ).images[0]
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
-    return image
+    shutil.rmtree("wandb")
+    return image, f"**WandB Run:** [{run_url}]({run_url})"
 
 
 gr.Interface(
     fn=generate_image,
     inputs=[
         gr.Textbox(label="WandB Project Name", value="Stable-Diffusion-XL"),
-        gr.Textbox(label="WandB API Key", type="password"),
+        gr.Textbox(label="WandB API Key (You can get one from https://wandb.ai/authorize)", type="password", value=None),
         gr.Textbox(
             label="Prompt",
             lines=3,
@@ -107,7 +114,84 @@ gr.Interface(
             label="Guidance Rescale Factor",
         ),
     ],
-    outputs="image",
-    title="Reproducible Stable Diffusion XL",
-    description="Generate images using Stable Diffusion.",
+    outputs=["image", "markdown"],
+    title="Reproducible Stable Diffusion-XL with WandB",
+    description="""
+    This space enables us to generate images using Stable Diffusion XL and track your experiments using Weights & Biases. In order to learn more able engineering better prompts for Stable Diffusion models and how to make your experiments reproducible using [Weights & Biases](https://wandb.ai/site), check out [this report](https://wandb.ai/geekyrakshit/diffusers-prompt-engineering/reports/A-Guide-to-Prompt-Engineering-for-Diffusion-Models--Vmlldzo1NzY4NzQ3).
+    This app is powered by [🤗 Diffusers](https://huggingface.co/docs/diffusers) and [Weights & Biases](https://wandb.ai/site). A **Weights & Biases API key is necessary** to run this app, which you can find at **[https://wandb.ai/authorize](https://wandb.ai/authorize)**.
+    """,
+    examples=[
+        [
+            "Stable-Diffusion-XL",
+            None,
+            "a woman in orange, extremely detailed digital painting, in the style of Fenghua Zhong and Ruan Jia and jeremy lipking and Peter Mohrbacher, mystical colors, rim light, beautiful Lighting, 8k, stunning scene, raytracing, octane, trending on artstation",
+            "blurry, plastic, grainy, duplicate, deformed , disfigured, poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limb, disconnected limb, mutated hands and fingers, text, name, signature, watermark, worst quality, jpeg artifacts, boring composition, uninteresting",
+            1024,
+            1024,
+            50,
+            5.0,
+            256174057,
+            0.0,
+        ],
+        [
+            "Stable-Diffusion-XL",
+            None,
+            "hyper realistic photograph, Kodak disposable photography of young queen sit on her throne, blonde hair, red lips, white skin, 8k",
+            None,
+            1024,
+            1024,
+            50,
+            5.0,
+            189708808,
+            0.0,
+        ],
+        [
+            "Stable-Diffusion-XL",
+            None,
+            "hyper realistic waist up portrait of a teen boy with messy dark blond hair, light brown eyes, smiling, dimples, textured skin, realistic features, imperfect skin, 85mm lens Sony Lens, bokeh, sharp focus, professional photography, warm lighting, soft lights, realistic sharp eyes, sharp focus, set a busy street in the evening as background",
+            None,
+            1024,
+            1024,
+            50,
+            5.0,
+            267243010,
+            0.0,
+        ],
+        [
+            "Stable-Diffusion-XL",
+            None,
+            "A Cinematic Film Still Food Photography of a Cocktail, shot on fujifilm xt4, Professional Advertisement Photography, Dynamic Effects, Foaming Gas, Epic, Beautiful Details, Mysterious micro effects, Highly Detailed",
+            None,
+            1024,
+            1024,
+            50,
+            5.0,
+            1073711987,
+            0.0,
+        ],
+        [
+            "Stable-Diffusion-XL",
+            None,
+            "hyper realistic photograph, photography of a children dressed like a gangster, 50 mm, film grain, Kodak portra 800",
+            None,
+            1024,
+            1024,
+            50,
+            5.0,
+            355517910,
+            0.0,
+        ],
+        [
+            "Stable-Diffusion-XL",
+            None,
+            "a girl as personification of chocolate cupcake",
+            "deformed, disfigured, poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limb, disconnected limb, mutated hands and fingers, blurry",
+            1024,
+            1024,
+            50,
+            5.0,
+            595687707,
+            0.0,
+        ],
+    ],
 ).launch(debug=True, max_threads=80)