Update app.py

app.py

@@ -21,6 +21,8 @@ from PIL import Image
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
+current_lora_scale = 1.0
+
 
 parser = argparse.ArgumentParser() 
 parser.add_argument("--port", type=int, default=7860, help="Port for the Gradio app")
@@ -31,8 +33,9 @@ parser.add_argument("--enable-model-cpu-offload", action="store_true", help="Ena
 args = parser.parse_args()
 
 pipe = FluxFillPipeline.from_pretrained(args.flux_path, torch_dtype=torch.bfloat16)
-pipe.load_lora_weights(args.lora_path)
-
+pipe.load_lora_weights(args.lora_path, adapter_name="icedit")
+pipe.set_adapters("icedit", 1.0)
+    
 if args.enable_model_cpu_offload:
     pipe.enable_model_cpu_offload() 
 else:
@@ -47,9 +50,19 @@ def infer(edit_images,
           height=1024, 
           guidance_scale=50, 
           num_inference_steps=28, 
+          lora_scale=1.0,
           progress=gr.Progress(track_tqdm=True)
 ):
     
+
+    global current_lora_scale
+    
+    if lora_scale != current_lora_scale:
+        print(f"\033[93m[INFO] LoRA scale changed from {current_lora_scale} to {lora_scale}, reloading LoRA weights\033[0m")
+        pipe.set_adapters("icedit", lora_scale)
+        current_lora_scale = lora_scale
+        
+    
     image = edit_images
         
     if image.size[0] != 512:
@@ -93,13 +106,14 @@ def infer(edit_images,
     index = len(os.listdir(args.output_dir))
     output_image.save(f"{args.output_dir}/result_{index}.png")
     
-    return (image, output_image), seed
+    return (image, output_image), seed, lora_scale
     
 # 新增的示例，将元组转换为列表
 new_examples = [
-    ['assets/girl.png', 'Make her hair dark green and her clothes checked.', 42],
-    ['assets/boy.png', 'Change the sunglasses to a Christmas hat.', 27440001],
-    ['assets/kaori.jpg', 'Make it a sketch.', 329918865]
+    ['assets/girl_3.jpg', 'Make it looks like a watercolor painting.', 0, 0.5],
+    ['assets/girl.png', 'Make her hair dark green and her clothes checked.', 42, 1.0],
+    ['assets/boy.png', 'Change the sunglasses to a Christmas hat.', 27440001, 1.0],
+    ['assets/kaori.jpg', 'Make it a sketch.', 329918865, 1.0]
 ]
 
 css = """
@@ -113,9 +127,11 @@ with gr.Blocks(css=css) as demo:
 
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""# IC-Edit
-**Image Editing is worth a single LoRA!** A demo for [IC-Edit](https://arxiv.org/pdf/2504.20690).
+**Image Editing is worth a single LoRA!** A demo for [IC-Edit](https://river-zhang.github.io/ICEdit-gh-pages/).
 More **open-source**, with **lower costs**, **faster speed** (it takes about 9 seconds to process one image), and **powerful performance**.
-For more details, check out our [Github Repository](https://github.com/River-Zhang/ICEdit) and [website](https://river-zhang.github.io/ICEdit-gh-pages/). If our project resonates with you or proves useful, we'd be truly grateful if you could spare a moment to give it a star.
+For more details, check out our [Github Repository](https://github.com/River-Zhang/ICEdit) and [arxiv paper](https://arxiv.org/pdf/2504.20690). If our project resonates with you or proves useful, we'd be truly grateful if you could spare a moment to give it a star.
+\n**👑 Feel free to share your results in this [Gallery](https://github.com/River-Zhang/ICEdit/discussions/21)!**
+\n🔥 New feature: Try **different LoRA scale**!
 """)
         with gr.Row():
             with gr.Column():
@@ -187,11 +203,17 @@ For more details, check out our [Github Repository](https://github.com/River-Zha
                     value=28,
                 )
 
-        # 添加 Gradio 示例组件
+            lora_scale = gr.Slider(
+                label="LoRA Scale",
+                minimum=0,
+                maximum=1.0,
+                step=0.01,
+                value=1.0,
+            )
         gr.Examples(
             examples=new_examples,
-            inputs=[edit_image, prompt, seed],
-            outputs=[result, seed],
+            inputs=[edit_image, prompt, seed, lora_scale],
+            outputs=[result, seed, lora_scale],
             fn=infer,
             cache_examples=False
         )
@@ -199,8 +221,8 @@ For more details, check out our [Github Repository](https://github.com/River-Zha
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,
-        inputs=[edit_image, prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
-        outputs=[result, seed]
+        inputs=[edit_image, prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, lora_scale],
+        outputs=[result, seed, lora_scale]
     )
 
 demo.launch(server_port=args.port)