Add pipeline tag and explicit GitHub link to model card

README.md

@@ -1,15 +1,18 @@
 ---
-library_name: transformers
-license: apache-2.0
-datasets:
-- ILSVRC/imagenet-1k
 base_model:
 - SixAILab/nepa-large-patch14-224
+datasets:
+- ILSVRC/imagenet-1k
+library_name: transformers
+license: apache-2.0
+pipeline_tag: image-classification
 ---
+
 # NEPA: Next-Embedding Prediction Makes Strong Vision Learners
 
 [![Paper](https://img.shields.io/badge/arXiv-Paper-b31b1b?logo=arxiv&logoColor=b31b1b)](https://arxiv.org/abs/2512.16922)
 [![Project Page](https://img.shields.io/badge/Project-Website-5B7493?logo=googlechrome&logoColor=5B7493)](https://sihanxu.github.io/nepa)
+[![Code](https://img.shields.io/badge/GitHub-Code-181717?logo=github&logoColor=white)](https://github.com/SihanXU/nepa)
 [![Hugging Model Card](https://img.shields.io/badge/huggingface-model:NEPA-lightblue)](https://huggingface.co/collections/SixAILab/nepa)
 
 This is a PyTorch/GPU re-implementation of Next-Embedding Prediction Makes Strong Vision Learners.