update README

README.md

@@ -4,7 +4,7 @@ language:
 - en
 base_model:
 - black-forest-labs/FLUX.1-Kontext-dev
-pipeline_tag: image-to-image
+pipeline_tag: text-to-image
 library_name: diffusers
 tags:
 - art
@@ -25,8 +25,9 @@ ContextGen: Contextual Layout Anchoring </br> for Identity-Consistent Multi-Inst
 <p align="center">
 <a href="https://github.com/nenhang/ContextGen"><img src="https://img.shields.io/badge/GitHub-Code-black" alt="GitHub Code"></a>
 <a href="https://nenhang.github.io/ContextGen"><img src="https://img.shields.io/badge/Project-Page-brightgreen" alt="Project Page"></a>
-<a href="https://arxiv.org/abs/2510.11000"><img src="https://img.shields.io/badge/arXiv-2510.11000-b31b1b" alt="Paper"></a>
+<a href="https://arxiv.org/abs/2510.11000"><img src="https://img.shields.io/badge/arXiv-Paper-b31b1b" alt="Paper"></a>
 <a href="https://github.com/nenhang/IMIG-Source"><img src="https://img.shields.io/badge/Dataset-Code-blue" alt="Dataset Code"></a>
+<a href="https://huggingface.co/datasets/ruihangxu/IMIG-100K"><img src="https://img.shields.io/static/v1?label=%F0%9F%A4%97%20Hugging%20Face&message=Dataset&color=red"></a>
 </p>
 
 <div align="center">