Update app.py

app.py

@@ -8,15 +8,15 @@ from SegBody import segment_body
 
 
 # Load models
-device = "cuda" if torch.cuda.is_available() else "cpu"
-vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16).to(device)
+
+vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipeline = AutoPipelineForInpainting.from_pretrained(
     "diffusers/stable-diffusion-xl-1.0-inpainting-0.1", 
     vae=vae, 
     torch_dtype=torch.float16, 
     variant="fp16", 
     use_safetensors=True
-).to(device)
+).to("cuda")
 pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter_sdxl.bin")
 
 # Function to process images