Update app.py

app.py

@@ -18,6 +18,7 @@ pipe.enable_vae_tiling()
 
 # Apply hidiffusion with a single line of code.
 apply_hidiffusion(pipe)
+apply_hidiffusion(pipe_15)
 
 @spaces.GPU
 def run_hidiffusion(prompt, negative_prompt, progress=gr.Progress(track_tqdm=True)):
@@ -27,10 +28,9 @@ def run_hidiffusion(prompt, negative_prompt, progress=gr.Progress(track_tqdm=Tru
 def run_hidiffusion_15(prompt, negative_prompt, progress=gr.Progress(track_tqdm=True)):
     return pipe_15(prompt, guidance_scale=7.5, height=1024, width=1024, eta=1.0, negative_prompt=negative_prompt, num_inference_steps=25).images[0]
 
-
 with gr.Blocks() as demo:
     gr.Markdown("# HiDiffusion Demo")
-    gr.Markdown("Make Stable Diffusion generated higher resolution images than what it was trained for")
+    gr.Markdown("Make diffusion models generate higher resolution images with Resolution-Aware U-Net & Multi-head Self-Attention. [Paper](https://huggingface.co/papers/2311.17528) | [Code](https://github.com/megvii-research/HiDiffusion)")
     with gr.Tab("SDXL in 2048x2048"):
         with gr.Row():
             prompt = gr.Textbox(label="Prompt")
@@ -41,8 +41,12 @@ with gr.Blocks() as demo:
             prompt_15 = gr.Textbox(label="Prompt")
             negative_prompt_15 = gr.Textbox(label="Negative Prompt")
         btn_15 = gr.Button("Run")
-    output = gr.Image()
-
+    output = gr.Image(label="Result")
+    gr.Examples(examples=[
+            ["Echoes of a forgotten song drift across the moonlit sea, where a ghost ship sails, its spectral crew bound to an eternal quest for redemption.", ""],
+            ["Roger rabbit as a real person, photorealistic, cinematic.", ""],
+            ["tanding tall amidst the ruins, a stone golem awakens, vines and flowers sprouting from the crevices in its body.",""]
+    ], inputs=[prompt, negative_prompt], outputs=[output])
     btn.click(fn=run_hidiffusion, inputs=[prompt, negative_prompt], outputs=[output])
     btn_15.click(fn=run_hidiffusion, inputs=[prompt_15, negative_prompt_15], outputs=[output])
 demo.launch()