Update app.py

app.py

@@ -12,7 +12,7 @@ import spaces
 import torch
 from diffusers import AutoencoderKL, DiffusionPipeline
 
-DESCRIPTION = "# SDXL"
+DESCRIPTION = "# RussianVibe XL 1.0"
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
 
@@ -27,12 +27,13 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 if torch.cuda.is_available():
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
     pipe = DiffusionPipeline.from_pretrained(
-        "stabilityai/stable-diffusion-xl-base-1.0",
+        "0x7o/RussianVibe-XL-v1.0",
         vae=vae,
         torch_dtype=torch.float16,
         use_safetensors=True,
         variant="fp16",
     )
+    pipeline.load_lora_weights("0x7o/RussianVibe-XL-v1.0")
     if ENABLE_REFINER:
         refiner = DiffusionPipeline.from_pretrained(
             "stabilityai/stable-diffusion-xl-refiner-1.0",
@@ -130,8 +131,7 @@ def generate(
 
 
 examples = [
-    "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
-    "An astronaut riding a green horse",
+    "The sun is setting through a window, casting a warm glow on the cityscape beyond. The sun casts a warm orange glow on the buildings in the distance, creating a beautiful and serene atmosphere.",
 ]
 
 with gr.Blocks(css="style.css") as demo: