Update app.py

app.py

@@ -9,8 +9,6 @@ from share_btn import community_icon_html, loading_icon_html, share_js
 from safetensors.torch import load_file
 from huggingface_hub import hf_hub_download
 
-unet = UNet2DConditionModel.from_config("SG161222/Realistic_Vision_V5.0_noVAE", subfolder="unet").to("cuda")
-unet.load_state_dict(load_file(hf_hub_download("ByteDance/SDXL-Lightning", "sdxl_lightning_2step_unet.safetensors"), device="cuda"))
 vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse")
 pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V5.0_noVAE",unet=unet,vae=vae).to("cuda")