Image Classification
Transformers
Safetensors
vit_nepa
nielsr HF Staff commited on
Commit
3491c1d
·
verified ·
1 Parent(s): a101892

Improve model card: Add image classification pipeline tag, GitHub link, and update project page URL

Browse files

This PR enhances the model card by making the following improvements:

- **Added `pipeline_tag: image-classification`**: This metadata ensures the model is correctly categorized on the Hugging Face Hub, improving its discoverability for users looking for image classification models.
- **Added GitHub repository link**: A direct link to the official GitHub repository (`https://github.com/SihanXU/nepa`) has been included as a badge for easy access to the code.
- **Updated Project Page URL**: The existing "Project Page" badge has been updated to point to the correct project website: `https://sihanxu.me/nepa`, resolving a factual discrepancy mentioned in the paper info.

The rest of the model card content, including the valuable "Quick Start" usage example and the arXiv paper link, has been preserved as it provides complete and accurate information.

Files changed (1) hide show
  1. README.md +8 -5
README.md CHANGED
@@ -1,16 +1,19 @@
1
  ---
2
- library_name: transformers
3
- license: apache-2.0
4
- datasets:
5
- - ILSVRC/imagenet-1k
6
  base_model:
7
  - SixAILab/nepa-base-patch14-224
 
 
 
 
 
8
  ---
 
9
  # NEPA: Next-Embedding Prediction Makes Strong Vision Learners
10
 
11
  [![Paper](https://img.shields.io/badge/arXiv-Paper-b31b1b?logo=arxiv&logoColor=b31b1b)](https://arxiv.org/abs/2512.16922)
12
- [![Project Page](https://img.shields.io/badge/Project-Website-5B7493?logo=googlechrome&logoColor=5B7493)](https://sihanxu.github.io/nepa)
13
  [![Hugging Model Card](https://img.shields.io/badge/huggingface-model:NEPA-lightblue)](https://huggingface.co/collections/SixAILab/nepa)
 
14
 
15
  This is a PyTorch/GPU re-implementation of Next-Embedding Prediction Makes Strong Vision Learners.
16
 
 
1
  ---
 
 
 
 
2
  base_model:
3
  - SixAILab/nepa-base-patch14-224
4
+ datasets:
5
+ - ILSVRC/imagenet-1k
6
+ library_name: transformers
7
+ license: apache-2.0
8
+ pipeline_tag: image-classification
9
  ---
10
+
11
  # NEPA: Next-Embedding Prediction Makes Strong Vision Learners
12
 
13
  [![Paper](https://img.shields.io/badge/arXiv-Paper-b31b1b?logo=arxiv&logoColor=b31b1b)](https://arxiv.org/abs/2512.16922)
14
+ [![Project Page](https://img.shields.io/badge/Project-Website-5B7493?logo=googlechrome&logoColor=5B7493)](https://sihanxu.me/nepa)
15
  [![Hugging Model Card](https://img.shields.io/badge/huggingface-model:NEPA-lightblue)](https://huggingface.co/collections/SixAILab/nepa)
16
+ [![Code](https://img.shields.io/badge/GitHub-Code-181717?logo=github)](https://github.com/SihanXU/nepa)
17
 
18
  This is a PyTorch/GPU re-implementation of Next-Embedding Prediction Makes Strong Vision Learners.
19