Update app.py

app.py

@@ -9,6 +9,7 @@ import uuid
 import json
 import os
 
+from huggingface_hub import hf_hub_download
 from diffusers import DiffusionPipeline
 from transformers import pipeline
 from PIL import Image
@@ -48,12 +49,16 @@ repo_default.load_lora_weights("nerijs/pixel-art-xl", adapter_name="pixel_base_2
 
 repo_pro = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, use_safetensors=True, add_watermarker=False)
 
+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_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,
+    "Pro": repo_pro.load_lora_weights(repo_pro_lora),
 }
 
 # Functions
@@ -92,7 +97,7 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
         guidance_set = 1.5
         repo.set_adapters(["pixel_base", "pixel_base_2"], adapter_weights=[1, 1])
     elif model == "Pro":   
-        steps_set = 25
+        steps_set = 8
         guidance_set = 3.5
     else:
         steps_set = 25