Update app.py

app.py

@@ -6,6 +6,7 @@ import torch
 from PIL import Image
 from SegBody import segment_body
 from SegCloth import segment_clothing
+import os
 
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipeline = AutoPipelineForInpainting.from_pretrained(os.environ.get('MODEL'), vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to("cuda")