Update app.py

app.py

@@ -4,12 +4,29 @@ import random
 import spaces
 import torch
 from diffusers import DiffusionPipeline
+from huggingface_hub import hf_hub_download
 
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe = DiffusionPipeline.from_pretrained("aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype).to(device)
 
+pipe.load_lora_weights(
+    hf_hub_download("aifeifei798/feifei-flux-lora-v1", "feifei.safetensors"),
+    adapter_name="feifei",
+)
+
+pipe.set_adapters(
+        ["feifei"],
+        adapter_weights=[0.65],
+    )
+pipe.fuse_lora(
+    adapter_name=["feifei"],
+    lora_scale=1.0,
+)
+pipe.unload_lora_weights()
+torch.cuda.empty_cache()
+
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
@@ -44,8 +61,8 @@ css="""
 with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(f"""# Shuttle 3.1 Aesthetic
-Shuttle 3.1 Aesthetic is a text-to-image AI model designed to create aesthetic, detailed and diverse images from textual prompts in just 4 steps. It offers enhanced performance in image quality, typography, understanding complex prompts, and resource efficiency.
+        gr.Markdown(f"""# DarkIdol-flux-FeiFei
+DarkIdol-flux-FeiFei is a text-to-image AI model designed to create aesthetic, detailed and diverse images from textual prompts in just 4-8 steps. It offers enhanced performance in image quality, typography, understanding complex prompts, and resource efficiency.
         """)
         
         with gr.Row():