Update app.py

app.py

@@ -27,6 +27,8 @@ model_file = hf_hub_download(
     filename="diffusion_pytorch_model_promax.safetensors",
 )
 state_dict = load_state_dict(model_file)
+# 获取 state_dict 中的所有键作为 loaded_keys
+loaded_keys = list(state_dict.keys())
 model = ControlNetModel_Union._load_pretrained_model(
     controlnet_model, 
     state_dict,