Update app.py

app.py

@@ -19,12 +19,16 @@ config_file = hf_hub_download(
 
 config = ControlNetModel_Union.load_config(config_file)
 controlnet_model = ControlNetModel_Union.from_config(config)
+
+
+# Load the state dictionary
 model_file = hf_hub_download(
     "xinsir/controlnet-union-sdxl-1.0",
     filename="diffusion_pytorch_model_promax.safetensors",
 )
+state_dict = load_state_dict(model_file)  # Define state_dict
 
-# First extract the keys from the state_dict
+# Then extract the keys from the state_dict
 loaded_keys = list(state_dict.keys())
 
 # Then call the method with the additional parameter