Update app.py

app.py

@@ -23,19 +23,12 @@ model_file = hf_hub_download(
     "xinsir/controlnet-union-sdxl-1.0",
     filename="diffusion_pytorch_model_promax.safetensors",
 )
-state_dict = load_state_dict(model_file)
 
-# **Fix**: pass loaded_keys into _load_pretrained_model
+sstate_dict = load_state_dict(model_file)
 model, _, _, _, _ = ControlNetModel_Union._load_pretrained_model(
-    controlnet_model,
-    state_dict,
-    loaded_keys,
-    model_file,
-    "xinsir/controlnet-union-sdxl-1.0",
+    controlnet_model, state_dict, model_file, "xinsir/controlnet-union-sdxl-1.0"
 )
-
 model.to(device="cuda", dtype=torch.float16)
-
 #----------------------
 
 vae = AutoencoderKL.from_pretrained(