Update app.py

app.py

@@ -6,9 +6,10 @@ import gradio as gr
 import torch
 import spaces
 
+model = "stabilityai/stable-diffusion-xl-base-1.0"
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
-scheduler = DDIMScheduler.from_pretrained(pretrain_model, subfolder="scheduler")
-pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", vae=vae, scheduler=scheduler, torch_dtype=torch.float16, use_safetensors=True, variant="fp16").to("cuda")
+scheduler = DDIMScheduler.from_pretrained(model, subfolder="scheduler")
+pipe = DiffusionPipeline.from_pretrained(model, vae=vae, scheduler=scheduler, torch_dtype=torch.float16, use_safetensors=True, variant="fp16").to("cuda")
 
 #pipe.enable_model_cpu_offload()
 #pipe.enable_vae_tiling()