Update app.py

app.py

@@ -7,7 +7,7 @@ from diffusers import DiffusionPipeline, AutoPipelineForText2Image
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "stabilityai/stable-diffusion-xl-base-1.0"
+model_repo_id = "ByteDance/SDXL-Lightning"
 
 if torch.cuda.is_available():
     torch_dtype = torch.float16
@@ -50,7 +50,7 @@ def infer(
     image = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
-        guidance_scale=0.0,  # SDXL-Turbo requires 0.0
+        guidance_scale=4,  # SDXL-Turbo requires 0.0
         num_inference_steps=max(1, min(4, num_inference_steps)),  # Clamp to 1-4
         width=width,
         height=height,