Update app.py

app.py

@@ -6,6 +6,8 @@ from diffusers.models import AutoencoderKL
 from diffusers import AutoPipelineForInpainting, UNet2DConditionModel
 import diffusers
 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", torch.float16)
 unet.load_state_dict(load_file(hf_hub_download("ByteDance/SDXL-Lightning", "sdxl_lightning_2step_unet.safetensors"), device="cuda"))