hub-task-tagger / README.md
davanstrien's picture
davanstrien HF Staff
README: public description, link the model and training script
1b57e46 verified
|
Raw History Blame Contribute Delete
599 Bytes
---
title: What is this dataset for?
emoji: 🏷️
colorFrom: gray
colorTo: blue
sdk: docker
app_port: 7860
pinned: false
short_description: Suggest Hub task tags from a dataset's first row
---
Suggests Hub `task_categories` for a dataset from its column names and first row, using a fine-tuned
[GLiNER2.5-base model](https://huggingface.co/davanstrien/hub-task-tagger-gliner2.5-base). Trained with
[`train-gliner2.py`](https://huggingface.co/datasets/uv-scripts/classification) on Hugging Face Jobs.
API: `POST /api/predict {"dataset_id": "owner/name"}` returns the top tags with probabilities.