Update space

app.py

@@ -15,6 +15,7 @@ pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
 # Adapt the InstructPix2Pix model using the LoRA parameters
 adapter_id = "latent-consistency/lcm-lora-sdv1-5"
 pipe.load_lora_weights(adapter_id)
+pipe.to('cuda')
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
@@ -48,30 +49,12 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(
             f"""
-        # ⚡ Invertible Consistency Distillation ⚡ 
-        # ⚡ Text-guided image editing with 8-step iCD-SD1.5 ⚡
-        This is a demo for [Invertible Consistency Distillation](https://yandex-research.github.io/invertible-cd/), 
-        a diffusion distillation method proposed in [Invertible Consistency Distillation for Text-Guided Image Editing in Around 7 Steps](https://arxiv.org/abs/2406.14539)
-        by [Yandex Research](https://github.com/yandex-research).
+        # ⚡ Instruct-p2p with Consistency Distillation⚡ 
         Currently running on {power_device}
         """
         )
         gr.Markdown(
-            "**Please** check the examples to catch the intuition behind the hyperparameters, which are quite important for successful editing. A short description: <br />1. *Dynamic guidance tau*. Controls the interval where guidance is applied: if t < tau, then guidance is turned on for t < tau."
-            " Lower tau values provide better reference preservation. We commonly use tau=0.6 and tau=0.8. <br />"
-            "2. *Cross replace steps (crs)* and *self replace steps (srs)*. Controls the time step interval " 
-            "where the cross- and self-attention maps are replaced. Higher values lead to better preservation of the reference image. "
-            "The optimal values depend on the particular image. " 
-            "Mostly, we use crs and srs from 0.2 to 0.6. <br />"
-            "3. *Amplify word* and *Amplify factor*. Define the word that needs to be enhanced in the edited image. <br />"
-            "4. *Blended word*. Specifies the object used for making local edits. That is, edit only selected objects. <br />"
-            "5. *Is replacement*. You can set True, if you replace only one word in the original prompt. But False also works in these cases."
-        )
-        gr.Markdown(
-            "Feel free to check out our [image generation demo](https://huggingface.co/spaces/dbaranchuk/iCD-image-generation) as well."
-        )
-        gr.Markdown(
-            "If you enjoy the space, feel free to give a ⭐ to the <a href='https://github.com/yandex-research/invertible-cd' target='_blank'>Github Repo</a>. [![GitHub Stars](https://img.shields.io/github/stars/yandex-research/invertible-cd?style=social)](https://github.com/yandex-research/invertible-cd)"
+            "If you enjoy the space, feel free to give a ⭐ to the <a href='https://github.com/yandex-research/invertible-cd' target='_blank'>Github Repo</a>. [![GitHub Stars](https://img.shields.io/github/stars/quickjkee/instruct-pix2pix-distill?style=social)](https://github.com/quickjkee/instruct-pix2pix-distill)"
         )
         with gr.Row():