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 = "ByteDance/SDXL-Lightning"
+model_repo_id = "stabilityai/stable-diffusion-xl-base-1.0"
 
 if torch.cuda.is_available():
     torch_dtype = torch.float16