Update app.py

app.py

@@ -29,12 +29,12 @@ model_file = hf_hub_download(
     "xinsir/controlnet-union-sdxl-1.0",
     filename="diffusion_pytorch_model_promax.safetensors",
 )
-
-        # Create patched version that includes the loaded_keys parameter
-        def patched_load_pretrained(self, pretrained_model_name_or_path, **kwargs):
-            if 'loaded_keys' not in kwargs:
-                kwargs['loaded_keys'] = set()
-            return original_load_pretrained(self, pretrained_model_name_or_path, **kwargs)
+        
+# Create patched version that includes the loaded_keys parameter
+def patched_load_pretrained(self, pretrained_model_name_or_path, **kwargs):
+    if 'loaded_keys' not in kwargs:
+        kwargs['loaded_keys'] = set()
+    return original_load_pretrained(self, pretrained_model_name_or_path, **kwargs)        
 
 state_dict = load_state_dict(model_file)
 model, _, _, _, _ = ControlNetModel_Union._load_pretrained_model(