Update app.py

app.py

@@ -7,6 +7,24 @@ import spaces
 
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 
+### RealVisXL V3 ###
+RealVisXLv3_pipe = DiffusionPipeline.from_pretrained(
+        "SG161222/RealVisXL_V3.0",
+        torch_dtype=torch.float16,
+        use_safetensors=True,
+        add_watermarker=False,
+        variant="fp16"
+    )
+RealVisXLv3_pipe.to("cuda")
+### RealVisXL V4 ###
+RealVisXLv4_pipe = DiffusionPipeline.from_pretrained(
+        "SG161222/RealVisXL_V4.0",
+        torch_dtype=torch.float16,
+        use_safetensors=True,
+        add_watermarker=False,
+        variant="fp16"
+    )
+RealVisXLv4_pipe.to("cuda")
 ### SDXL Turbo #### 
 pipe_turbo = StableDiffusionXLPipeline.from_pretrained("stabilityai/sdxl-turbo",
                                                        vae=vae,
@@ -64,6 +82,10 @@ def run_comparison(prompt, progress=gr.Progress(track_tqdm=True)):
     yield image_turbo, image_lightning, None
     image_hyper=pipe_hyper(prompt=prompt, num_inference_steps=1, guidance_scale=0, timesteps=[800]).images[0]
     yield image_turbo, image_lightning, image_hyper
+    image_turbo=RealVisXLv3_pipe(prompt=prompt, num_inference_steps=8, guidance_scale=0).images[0]
+    yield image_turbo, None, None
+    image_lightning=RealVisXLv4_pipe(prompt=prompt, num_inference_steps=8, guidance_scale=0).images[0]
+    yield image_turbo, image_lightning, None
 
 examples = ["A dignified beaver wearing glasses, a vest, and colorful neck tie.",
 "The spirit of a tamagotchi wandering in the city of Barcelona",
@@ -89,7 +111,13 @@ with gr.Blocks() as demo:
         with gr.Column():
             image_hyper = gr.Image(label="Hyper SDXL")
             gr.Markdown("## [Hyper SDXL](https://huggingface.co/ByteDance/Hyper-SD)")
-    image_outputs = [image_turbo, image_lightning, image_hyper]
+        with gr.Column():
+            image_r3 = gr.Image(label="RealVisXL V3")
+            gr.Markdown("## [RealVisXL V3](https://huggingface.co)")
+        with gr.Column():
+            image_r4 = gr.Image(label="RealVisXL V4")
+            gr.Markdown("## [RealVisXL V3](https://huggingface.co)")
+    image_outputs = [image_turbo, image_lightning, image_hyper, image_r3, image_4]
     gr.on(
         triggers=[prompt.submit, run.click],
         fn=run_comparison,