Update app.py

app.py

@@ -22,8 +22,7 @@ opts = {
 step_loaded = None
 # unet = UNet2DConditionModel.from_config(base, subfolder="unet").to(device, dtype)
 # unet.load_state_dict(load_file(hf_hub_download(repo, opts["4 Steps"][0]), device=device))
-unet = None
-pipe = StableDiffusionXLPipeline.from_pretrained(base, unet=unet, torch_dtype=dtype, variant="fp16").to(device)
+pipe = StableDiffusionXLPipeline.from_pretrained(base, torch_dtype=dtype, variant="fp16").to(device)
 
 @spaces.GPU(enable_queue=True)
 def generate_image(prompt, option):