Update app.py

app.py

@@ -12,8 +12,8 @@ import torch
 from diffusers import AutoencoderKL, StableDiffusionXLPipeline
 import uuid
 
-DESCRIPTION = '''# Segmind Stable Diffusion: SSD-1B
-#### [Segmind's SSD-1B](https://huggingface.co/segmind/SSD-1B) is a distilled, 50% smaller version of SDXL, offering up to 60% speedup
+DESCRIPTION = '''# Segmind-Vega
+#### [Segmind-Vega](https://huggingface.co/segmind/Segmind-Vega) is a distilled, 70% smaller version of SDXL, offering up to 100% speedup
 '''
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
@@ -95,7 +95,7 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str
 if torch.cuda.is_available():
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
     pipe = StableDiffusionXLPipeline.from_pretrained(
-        "segmind/SSD-1B",
+        "segmind/Segmind-Vega",
         vae=vae,
         torch_dtype=torch.float16,
         use_safetensors=True,