Update app.py

app.py

@@ -15,22 +15,30 @@ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
 
 lora_repo = "kaytoo2022/cara-the-cavapoo-flux"
 trigger_word = ""  # Leave trigger_word blank if not used.
-pipe.load_lora_weights(lora_repo)
+pipe.load_lora_weights(lora_repo, adapter_name='cara')
 
-# lora_ghibli = 'alvarobartt/ghibli-characters-flux-lora'
-# pipe.load_lora_weights(lora_ghibli)
+# ghibsky
+lora_repo_2 = "aleksa-codes/flux-ghibsky-illustration"
+pipe.load_lora_weights(lora_repo_2, adapter_name='lora_2')
+
+lora_repo_3 = "Datou1111/shou_xin"
+pipe.load_lora_weights(lora_repo_3, adapter_name='lora_3')
+
+pipe.set_adapters(["cara", "lora_2", "lora_3"], adapter_weights=[0.85, 0.0, 0.0])
 
 pipe.to("cuda")
 
 MAX_SEED = 2**32-1
 
 @spaces.GPU()
-def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
+def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, lora_scale_2, lora_scale_3, progress=gr.Progress(track_tqdm=True)):
     # Set random seed for reproducibility
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device="cuda").manual_seed(seed)
 
+    pipe.set_adapters(["cara", "lora_2", "lora_3"], adapter_weights=[lora_scale, lora_scale_2, lora_scale_3])
+
     # Update progress bar (0% saat mulai)
     progress(0, "Starting image generation...")
 
@@ -48,7 +56,7 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
         width=width,
         height=height,
         generator=generator,
-        joint_attention_kwargs={"scale": lora_scale},
+        # joint_attention_kwargs={"scale": lora_scale},
     ).images[0]
 
     # Final update (100%)
@@ -57,19 +65,21 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
     yield image, seed
 
 # Example cached image and settings
-example_image_path = "cara-shark.jpg"  # Replace with the actual path to the example image
-example_prompt = """Cara the Cavapoo showing off her cool new t shirt at the beach, a shark is jumping out of the water in the background"""
+example_image_path = "bella_space.jpeg"  # Replace with the actual path to the example image
+example_prompt = """A portrait picture of b3lla dog in an astronaut outfit. Planets are visible in the background"""
 example_cfg_scale = 3.2
 example_steps = 32
 example_width = 1152
 example_height = 896
 example_seed = 3981632454
 example_lora_scale = 0.85
+example_lora_scale_2 = 0.0
+example_lora_scale_3 = 0.0
 
 def load_example():
     # Load example image from file
     example_image = Image.open(example_image_path)
-    return example_prompt, example_cfg_scale, example_steps, True, example_seed, example_width, example_height, example_lora_scale, example_image
+    return example_prompt, example_cfg_scale, example_steps, True, example_seed, example_width, example_height, example_lora_scale, example_lora_scale_2, example_lora_scale_3, example_image
 
 with gr.Blocks() as app:
     gr.Markdown("# Flux Lora Image Generator")
@@ -84,16 +94,18 @@ with gr.Blocks() as app:
             randomize_seed = gr.Checkbox(True, label="Randomize seed")
             seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=example_seed)
             lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=1, step=0.01, value=example_lora_scale)
+            lora_scale_2 = gr.Slider(label="LoRA Scale (GhibSky)", minimum=0, maximum=1, step=0.01, value=example_lora_scale_2)
+            lora_scale_3 = gr.Slider(label="LoRA Scale (Sketch)", minimum=0, maximum=1, step=0.01, value=example_lora_scale_3)
         with gr.Column(scale=1):
             result = gr.Image(label="Generated Image")
-            gr.Markdown("Generate images using RealismLora and a text prompt.\n[[non-commercial license, Flux.1 Dev](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md)]")
+            gr.Markdown("Generate images using Flux and a text prompt.\nUse `b3lla dog` in the prompt to trigger generating an image of Bella the dog.\n[[non-commercial license, Flux.1 Dev](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md)]")
 
     # Automatically load example data and image when the interface is launched
-    app.load(load_example, inputs=[], outputs=[prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, result])
+    app.load(load_example, inputs=[], outputs=[prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, lora_scale_2, lora_scale_3, result])
     
     generate_button.click(
         run_lora,
-        inputs=[prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale],
+        inputs=[prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, lora_scale_2, lora_scale_3],
         outputs=[result, seed]
     )