Update README.md

README.md

@@ -12,6 +12,14 @@ tags:
 ---
 **LeX-Enhancer** is a lightweight prompt enhancement model distilled from DeepSeek-R1. Specifically, we collect **60,856 prompt pairs** before and after DeepSeek-R1 enhancement, and fine-tune a Deepseek-R1-Distilled-Qwen-14B model using LoRA to replicate the detailed prompting capabilities of R1. This enables efficient, large-scale generation of high-quality, visually grounded prompts.
 
+```shell
+**Input: What you would like to paint in the text prompt.**
+A thank you card with the words very much, with the text on it: \"VERY\" in black, \"MUCH\" in yellow.
+
+**Output: Enhancer text prompt.**
+The image features a rectangular thank you card with a muted, warm-toned background in a soft gradient of creamy beige and blush pink, providing a subtle textured finish that adds depth without overwhelming the central elements. Centered near the top of the card, the word "VERY" is rendered in thick, uppercase black lettering with a clean sans-serif font, slightly curved to follow the natural flow of the composition. Directly beneath it, the word "MUCH" appears in a vibrant, sunny-yellow hue, matching the same bold font style but slightly larger in size, positioned to align neatly with the lower edge of "VERY," creating a balanced hierarchy. Both texts are outlined with faint drop shadows, enhancing readability against the softly lit backdrop. The cards’ edges are crisp, with minimalistic borders in a lighter shade of the background tones, and the overall layout is symmetrical, exuding a polished, contemporary aesthetic. Gentle ambient lighting casts soft, diffused shadows around the card’s corners, suggesting a lightly textured surface underneath, while the absence of decorative embellishments keeps the focus on the typography. The color palette harmonizes warmth and neutrality, ensuring the text remains the focal point while maintaining a serene, approachable ambiance.
+```
+
 Use this code for inference:
 ```python
 import torch, os
@@ -23,8 +31,9 @@ SYSTEM_TEMPLATE = (
     "The reasoning process and answer are enclosed within <think> </think> and <answer> </answer> tags, respectively, i.e., "
     "<think> reasoning process here </think> <answer> answer here </answer>."
 )
-
 model_path = 'X-ART/LeX-Enhancer'
+
+# Change to what you want to draw in next line
 simple_caption = "A thank you card with the words very much, with the text on it: \"VERY\" in black, \"MUCH\" in yellow."
 
 def create_chat_template(user_prompt):