Update app.py

app.py

@@ -52,13 +52,14 @@ repo_pro = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", tor
 repo_pro_dir = "alimama-creative/FLUX.1-Turbo-Alpha"
 repo_pro_ckpt = "diffusion_pytorch_model.safetensors"
 repo_pro_lora = hf_hub_download(repo_pro_dir, repo_pro_ckpt)
+repo_pro.load_lora_weights(repo_pro_lora)
 
 repo_customs = {
     "Default": repo_default,
     "Realistic": DiffusionPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
     "Anime": DiffusionPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
     "Pixel": repo_default,
-    "Pro": repo_pro.load_lora_weights(repo_pro_lora),
+    "Pro": repo_pro
 }
 
 # Functions