Update app.py

app.py

@@ -23,19 +23,21 @@ pipe.load_lora_weights(lora_ghibli, adapter_name='ghibli')
 lora_repo_anime = "aleksa-codes/flux-ghibsky-illustration"
 pipe.load_lora_weights(lora_repo_anime, adapter_name='ghibsky')
 
-pipe.set_adapters(["kaytoo", "ghibli", "ghibsky"], adapter_weights=[0.85, 0.65, 0.65])
+pipe.set_adapters(["kaytoo", "ghibli", "ghibsky"], adapter_weights=[0.6, 0.35, 0.55])
 
 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_ghibli, lora_scale_ghibsky, 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(["kaytoo", "ghibli", "ghibsky"], adapter_weights=[lora_scale, lora_scale_2])
+
     # Update progress bar (0% saat mulai)
     progress(0, "Starting image generation...")
 
@@ -53,7 +55,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%)
@@ -102,7 +104,7 @@ with gr.Blocks() as app:
     
     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_ghibli, lora_scale_ghibsky],
         outputs=[result, seed]
     )