Update app.py

app.py

@@ -8,6 +8,7 @@ import os
 from huggingface_hub import hf_hub_download
 import torch
 from diffusers import DiffusionPipeline
+from huggingface_hub import hf_hub_download
 
 # Constants
 MAX_SEED = np.iinfo(np.int32).max
@@ -18,6 +19,8 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev",
                                          custom_pipeline="pipeline_flux_rf_inversion",
                                           torch_dtype=torch.bfloat16)
+pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), lora_scale=0.125)
+pipe.fuse_lora(lora_scale=0.125)
 pipe.to(DEVICE)
 
 def reset_do_inversion():
@@ -86,10 +89,11 @@ with gr.Blocks(css=css) as demo:
     do_inversion = gr.State(False)
     
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(f"""# RF inversion with FLUX.1 [dev] 🖌️🏞️
-Edit real images with Flux, based on the algorithm proposed in [*Semantic Image Inversion and Editing using
+        gr.Markdown(f"""# RF inversion 🖌️🏞️
+### Edit real images with FLUX.1 [dev]
+based on the algorithm proposed in [*Semantic Image Inversion and Editing using
 Stochastic Rectified Differential Equations*](https://rf-inversion.github.io/data/rf-inversion.pdf)
-[[non-commercial license](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md)] [[project page](https://rf-inversion.github.io/] [[arxiv](https://arxiv.org/pdf/2410.10792)]
+[[non-commercial license](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md)] [[project page](https://rf-inversion.github.io/) [[arxiv](https://arxiv.org/pdf/2410.10792)]
         """)
         
         with gr.Row():
@@ -132,8 +136,8 @@ Stochastic Rectified Differential Equations*](https://rf-inversion.github.io/dat
                     label="num inversion steps",
                     minimum=1,
                     maximum=50,
-                    step=32,
-                    value=28,
+                    step=1,
+                    value=8,
                 )
             
             with gr.Row():