Update app.py

app.py

@@ -28,7 +28,7 @@ USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
 ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-pipe = AutoPipelineForText2Image.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0',use_safetensors=True)# torch_dtype=torch.float16, variant="safetensors")
+pipe = AutoPipelineForText2Image.from_pretrained('Lykon/dreamshaper-xl-1-0',use_safetensors=True)# torch_dtype=torch.float16, variant="safetensors")
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, use_karras_sigmas=True, algorithm_type="sde-dpmsolver++")