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
metadata
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. Trained with train-gliner2.py on Hugging Face Jobs.

API: POST /api/predict {"dataset_id": "owner/name"} returns the top tags with probabilities.