nielsr HF Staff commited on
Commit
fc06ce2
·
verified ·
1 Parent(s): bdb17d3

Correct pipeline tag and add link to Github repo

Browse files

This PR updates the `pipeline_tag` to `text-to-image` to more accurately reflect the model's function as a prompt enhancer for text-to-image generation and adds a link to the Github repository.

Files changed (1) hide show
  1. README.md +14 -9
README.md CHANGED
@@ -1,14 +1,14 @@
1
  ---
 
 
 
 
2
  library_name: transformers
3
  license: mit
4
- datasets:
5
- - X-ART/LeX-R1-60K
6
- base_model:
7
- - deepseek-ai/DeepSeek-R1-Distill-Qwen-14B
8
- pipeline_tag: text-generation
9
  tags:
10
- - art
11
- - text-rendering
12
  ---
13
 
14
  # 🎨 LeX-Enhancer: Visual Prompt Generator
@@ -71,7 +71,9 @@ def create_user_prompt(simple_caption):
71
  "5. The detailed caption should be human readable and fluent. "
72
  "6. Avoid using vague expressions such as \"may be\" or \"might be\"; the generated caption must be in a definitive, narrative tone. "
73
  "7. Do not use negative sentence structures, such as \"there is nothing in the image,\" etc. The entire caption should directly describe the content of the image. "
74
- "8. The entire output should be limited to 200 words.\n\n"
 
 
75
  f"SIMPLE CAPTION: {simple_caption}"
76
  )
77
 
@@ -96,4 +98,7 @@ output = model.generate(
96
  )
97
 
98
  print("*" * 80)
99
- # Output will stream via TextStreamer
 
 
 
 
1
  ---
2
+ base_model:
3
+ - deepseek-ai/DeepSeek-R1-Distill-Qwen-14B
4
+ datasets:
5
+ - X-ART/LeX-R1-60K
6
  library_name: transformers
7
  license: mit
8
+ pipeline_tag: text-to-image
 
 
 
 
9
  tags:
10
+ - art
11
+ - text-rendering
12
  ---
13
 
14
  # 🎨 LeX-Enhancer: Visual Prompt Generator
 
71
  "5. The detailed caption should be human readable and fluent. "
72
  "6. Avoid using vague expressions such as \"may be\" or \"might be\"; the generated caption must be in a definitive, narrative tone. "
73
  "7. Do not use negative sentence structures, such as \"there is nothing in the image,\" etc. The entire caption should directly describe the content of the image. "
74
+ "8. The entire output should be limited to 200 words.
75
+
76
+ "
77
  f"SIMPLE CAPTION: {simple_caption}"
78
  )
79
 
 
98
  )
99
 
100
  print("*" * 80)
101
+ # Output will stream via TextStreamer
102
+ ```
103
+
104
+ Github repository: https://github.com/zhaoshitian/LeX-Art