Correct pipeline tag and add link to Github repo (#1)

- Correct pipeline tag and add link to Github repo (fc06ce2fdfd8774759bbe413dc9fcec1a3000d5f)


Co-authored-by: Niels Rogge <[email protected]>

README.md

@@ -1,14 +1,14 @@
 ---
+base_model:
+- deepseek-ai/DeepSeek-R1-Distill-Qwen-14B
+datasets:
+- X-ART/LeX-R1-60K
 library_name: transformers
 license: mit
-datasets:
-  - X-ART/LeX-R1-60K
-base_model:
-  - deepseek-ai/DeepSeek-R1-Distill-Qwen-14B
-pipeline_tag: text-generation
+pipeline_tag: text-to-image
 tags:
-  - art
-  - text-rendering
+- art
+- text-rendering
 ---
 
 [![lexart](https://img.shields.io/badge/Paper-LeXArt-2b9348.svg?logo=arXiv)](https://arxiv.org/abs/2503.21749)
@@ -73,7 +73,9 @@ def create_user_prompt(simple_caption):
         "5. The detailed caption should be human readable and fluent. "
         "6. Avoid using vague expressions such as \"may be\" or \"might be\"; the generated caption must be in a definitive, narrative tone. "
         "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. "
-        "8. The entire output should be limited to 200 words.\n\n"
+        "8. The entire output should be limited to 200 words.
+
+"
         f"SIMPLE CAPTION: {simple_caption}"
     )
 
@@ -98,4 +100,7 @@ output = model.generate(
 )
 
 print("*" * 80)
-# Output will stream via TextStreamer
+# Output will stream via TextStreamer
+```
+
+Github repository: https://github.com/zhaoshitian/LeX-Art