Update app.py

app.py

@@ -1,10 +1,9 @@
 import gradio as gr
-from diffusers import AutoPipelineForText2Image
+from diffusers import DiffusionPipeline
 import torch
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
-pipe.to(device)
+pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
+pipe.enable_model_cpu_offload()
 
 def infer(prompt):