Update app.py

app.py

@@ -40,6 +40,16 @@ RealVisXLv4_pipe = DiffusionPipeline.from_pretrained(
         variant="fp16"
     )
 RealVisXLv4_pipe.to("cuda")
+## playground V2.5##
+play_pipe = DiffusionPipeline.from_pretrained(
+        "playgroundai/playground-v2.5-1024px-aesthetic",
+        torch_dtype=torch.float16,
+        vae=vae,
+        use_safetensors=True,
+        add_watermarker=False,
+        variant="fp16"
+    )
+play_pipe_pipe.to("cuda")
 
 @spaces.GPU
 def run_comparison(prompt: str,
@@ -81,7 +91,18 @@ def run_comparison(prompt: str,
     ).images
     image_paths_r4 = [save_image(img) for img in image_r4]
 
-    return image_paths_r3, image_paths_r4, seed
+    play_pipe = play_pipe(prompt=prompt,
+        negative_prompt=negative_prompt,
+        width=width,
+        height=height,
+        guidance_scale=guidance_scale,
+        num_inference_steps=num_inference_steps,
+        num_images_per_prompt=num_images_per_prompt,
+        cross_attention_kwargs={"scale": 0.65},
+        output_type="pil",
+    ).images
+    image_paths_play_pipe = [save_image(img) for img in image_r4]
+    return image_paths_r3, image_paths_r4,image_paths_play_pipe, seed
 
 examples = ["A dignified beaver wearing glasses, a vest, and colorful neck tie.",
 "The spirit of a tamagotchi wandering in the city of Barcelona",
@@ -163,7 +184,10 @@ with gr.Blocks() as demo:
         with gr.Column():
             image_r4 = gr.Gallery(label="RealVisXL V4",columns=1, preview=True,)
             gr.Markdown("## [RealVisXL V3](https://huggingface.co)")
-    image_outputs = [image_r3, image_r4]
+        with gr.Column():
+            play_pipe = gr.Gallery(label="RealVisXL V4",columns=1, preview=True,)
+            gr.Markdown("## [RealVisXL V3](https://huggingface.co)")    
+    image_outputs = [image_r3, image_r4, play_pipe]
     gr.on(
         triggers=[prompt.submit, run.click],
         fn=run_comparison,