Update README.md

README.md

@@ -85,52 +85,16 @@ widget:
     margin-bottom: 0em;
   }
 
-  .overlay {
-    position: absolute;
-    bottom: 0;
-    left: 0;
-    right: 0;
-    color: white;
-    width: 100%;
-    height: 40%;
-    display: flex;
-    flex-direction: column;
-    justify-content: center;
-    align-items: center;
-    font-size: 1em;
-    font-style: bold;
-    text-align: center;
-    opacity: 0;
-    /* Keep the text fully opaque */
-    background: linear-gradient(0deg, rgba(0, 0, 0, 0.8) 60%, rgba(0, 0, 0, 0) 100%);
-    transition: opacity .5s;
-  }
-
-  .custom-image-container:hover .overlay {
-    opacity: 1;
-    /* Make the overlay always visible */
+  .nsfw-filter {
+    filter: blur(8px); /* Apply a blur effect */
+    transition: filter 0.3s ease; /* Smooth transition for the blur effect */
   }
 
-  .overlay-text {
-    background: linear-gradient(45deg, #7ed56f, #28b485);
-    -webkit-background-clip: text;
-    color: transparent;
-    /* Fallback for browsers that do not support this effect */
-    text-shadow: 2px 2px 4px rgba(0, 0, 0, 0.7);
-    /* Enhanced text shadow for better legibility */
-  }
-
-  .overlay-subtext {
-    font-size: 0.75em;
-    margin-top: 0.5em;
-    font-style: italic;
-  }
-
-  .overlay,
-  .overlay-subtext {
-    text-shadow: 2px 2px 4px rgba(0, 0, 0, 0.5);
+  .custom-image-container:hover .nsfw-filter {
+    filter: none; /* Remove the blur effect on hover */
   }
 </style>
+
 <h1 class="title">
   <span>Style Enhancer XL LoRA</span>
 </h1>
@@ -138,34 +102,25 @@ widget:
   <tr>
     <td>
       <div class="custom-image-container">
-        <a href="https://huggingface.co/Linaqruf/animagine-xl/blob/main/sample_images/image1.png">
-          <img class="custom-image" src="https://huggingface.co/Linaqruf/animagine-xl/resolve/main/sample_images/image1.png" alt="sample1">
-          <div class="overlay"> Twilight Whispers <div class="overlay-subtext">"A serene gaze into the dusky lights"</div>
-          </div>
-        </a>
+        <img class="custom-image" src="https://cdn-uploads.huggingface.co/production/uploads/6365c8dbf31ef76df4042821/kMqCcN3CpMPHO1qyEgnGk.png" alt="sample1">
       </div>
       <div class="custom-image-container">
-        <a href="https://huggingface.co/Linaqruf/animagine-xl/blob/main/sample_images/image4.png">
-          <img class="custom-image" src="https://huggingface.co/Linaqruf/animagine-xl/resolve/main/sample_images/image4.png" alt="sample4">
-          <div class="overlay"> Bloom of Youth <div class="overlay-subtext">"Amidst the dance of petals and sunbeams"</div>
-          </div>
-        </a>
+        <img class="custom-image" src="https://cdn-uploads.huggingface.co/production/uploads/6365c8dbf31ef76df4042821/pfLIDf7ZX6WJHgTlfqiQk.png" alt="sample4">
       </div>
     </td>
     <td>
       <div class="custom-image-container">
-        <a href="https://huggingface.co/Linaqruf/animagine-xl/blob/main/sample_images/image2.png">
-          <img class="custom-image" src="https://huggingface.co/Linaqruf/animagine-xl/resolve/main/sample_images/image2.png" alt="sample2">
-          <div class="overlay"> Starry-eyed Dreams <div class="overlay-subtext">"Lost in the constellation of imagination"</div>
-          </div>
-        </a>
+        <img class="custom-image" src="https://cdn-uploads.huggingface.co/production/uploads/6365c8dbf31ef76df4042821/s5ZCIaETb_eKijgbLbXwU.png" alt="sample2">
       </div>
       <div class="custom-image-container">
-        <a href="https://huggingface.co/Linaqruf/animagine-xl/blob/main/sample_images/image3.png">
-          <img class="custom-image" src="https://huggingface.co/Linaqruf/animagine-xl/resolve/main/sample_images/image3.png" alt="sample3">
-          <div class="overlay"> Sunset Serenade <div class="overlay-subtext">"A fiery glance that kindles the heart"</div>
-          </div>
-        </a>
+        <img class="custom-image" src="https://cdn-uploads.huggingface.co/production/uploads/6365c8dbf31ef76df4042821/qjC5jKExA_JE5BuQJ6-Ue.png" alt="sample3">
+    </td>
+    <td>
+      <div class="custom-image-container">
+        <img class="custom-image nsfw-filter" src="https://cdn-uploads.huggingface.co/production/uploads/6365c8dbf31ef76df4042821/1vvUY1qbgot5np4CPmORh.png" alt="sample1">
+      </div>
+      <div class="custom-image-container">
+        <img class="custom-image" src="https://cdn-uploads.huggingface.co/production/uploads/6365c8dbf31ef76df4042821/7OQJdRKKpZponZiZ8cOre.png" alt="sample4">
       </div>
     </td>
   </tr>
@@ -175,38 +130,33 @@ widget:
 
 ## Overview
 
-**Style Enhancer XL LoRA** is a high resolution, LoRA adapter for Animagine XL 2.0. The model has been fine-tuned using a curated dataset of superior-quality anime-style images.
-
-Like other anime-style Stable Diffusion models, it also supports Danbooru tags to generate images.
-
-e.g. _**face focus, cute, masterpiece, best quality, 1girl, green hair, sweater, looking at viewer, upper body, beanie, outdoors, night, turtleneck**_
+**Style Enhancer XL LoRA** is an advanced, high-resolution LoRA (Low-Rank Adaptation) adapter designed to enhance the capabilities of Animagine XL 2.0. This innovative model excels in fine-tuning and refining anime-style images, producing unparalleled quality and detail. It seamlessly integrates with the Stable Diffusion XL framework, and uniquely supports Danbooru tags for precise and creative image generation.
 
+Example tags include _**face focus, cute, masterpiece, best quality, 1girl, green hair, sweater, looking at viewer, upper body, beanie, outdoors, night, turtleneck**_.
 
 <hr>
 
 ## Model Details
 
 - **Developed by:** [Linaqruf](https://github.com/Linaqruf)
-- **Model type:** LoRA adapter
-- **Model Description:** This is a small model that should be used with big model and can be used to generate and modify high quality anime-themed images based on text prompts. This adapter can enhance Animagine XL 2.0 style and also get `old-school` SD 1.5 anime model artstyle.
+- **Model type:** LoRA adapter for Stable Diffusion XL
+- **Model Description:** A compact yet powerful adapter designed to augment and enhance the output of large models like Animagine XL 2.0. This adapter not only improves the style and quality of anime-themed images but also allows users to recreate the distinct 'old-school' art style of SD 1.5. It's the perfect tool for generating high-fidelity, anime-inspired visual content.
 - **License:** [CreativeML Open RAIL++-M License](https://huggingface.co/stabilityai/stable-diffusion-2/blob/main/LICENSE-MODEL)
 - **Finetuned from model:** [Animagine XL 2.0](https://huggingface.co/Linaqruf/animagine-xl-2.0)
 
 <hr>
 
-## 🧨 Diffusers 
+## 🧨 Diffusers Installation
 
-Make sure to upgrade diffusers to >= 0.23.0:
-```
+Ensure the installation of the latest `diffusers` library, along with other essential packages:
+
+```bash
 pip install diffusers --upgrade
+pip install transformers accelerate safetensors
 ```
 
-In addition make sure to install `transformers`, `safetensors`, `accelerate` as well as the invisible watermark:
-```
-pip install invisible_watermark transformers accelerate safetensors
-```
+The following Python script demonstrates how to utilize the Style Enhancer XL LoRA with Animagine XL 2.0. The default scheduler is EulerAncestralDiscreteScheduler, but it can be explicitly defined for clarity.
 
-Running the pipeline (The default scheduler for Animagine XL 2.0 is **EulerAncestralDiscreteScheduler** but you may also declare it in the code if you want to make sure)*:
 ```py
 import torch
 from diffusers import (
@@ -215,14 +165,17 @@ from diffusers import (
     AutoencoderKL
 )
 
+# Initialize LoRA model and weights
 lora_model_id = "Linaqruf/style-enhancer-xl-lora"
 lora_filename = "style-enhancer-xl.safetensors"
 
+# Load VAE component
 vae = AutoencoderKL.from_pretrained(
     "madebyollin/sdxl-vae-fp16-fix", 
     torch_dtype=torch.float16
 )
 
+# Configure the pipeline
 pipe = StableDiffusionXLPipeline.from_pretrained(
     "Linaqruf/animagine-xl-2.0", 
     vae=vae,
@@ -230,13 +183,14 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
     use_safetensors=True, 
     variant="fp16"
 )
-
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 pipe.to('cuda')
 
+# Load and fuse LoRA weights
 pipe.load_lora_weights(lora_model_id, weight_name=lora_filename)
 pipe.fuse_lora(lora_scale=0.6)
 
+# Define prompts and generate image
 prompt = "face focus, cute, masterpiece, best quality, 1girl, green hair, sweater, looking at viewer, upper body, beanie, outdoors, night, turtleneck"
 negative_prompt = "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry"
 
@@ -249,9 +203,10 @@ image = pipe(
     num_inference_steps=50
 ).images[0]
 
+# Unfuse LoRA before saving the image
 pipe.unfuse_lora()
-
 image.save("anime_girl.png")
+
 ```
 <hr>