Update app.py

app.py

@@ -13,10 +13,10 @@ SAFETY_CHECKER = os.environ.get("SAFETY_CHECKER", "0") == "1"
 base = "stabilityai/stable-diffusion-xl-base-1.0"
 repo = "ByteDance/SDXL-Lightning"
 checkpoints = {
-    "1-Step" : ["sdxl_lightning_1step_unet_x0.safetensors", 1],
-    "2-Step" : ["sdxl_lightning_2step_unet.safetensors", 2],
-    "4-Step" : ["sdxl_lightning_4step_unet.safetensors", 4],
-    "8-Step" : ["sdxl_lightning_8step_unet.safetensors", 8],
+    "Warp 1": ["sdxl_lightning_1step_unet_x0.safetensors", 1],
+    "Warp 2": ["sdxl_lightning_2step_unet.safetensors", 2],
+    "Warp 4": ["sdxl_lightning_4step_unet.safetensors", 4],
+    "Warp 8": ["sdxl_lightning_8step_unet.safetensors", 8],
 }
 loaded = None