Load your dataset to Argilla

!pip install argilla datasets

import argilla as rg

HF_TOKEN = "..."  # only for private spaces

client = rg.Argilla(
    api_url="...",
    api_key="...",
    headers={"Authorization": f"Bearer {HF_TOKEN}"},  # only for private spaces
)

from datasets import load_dataset

data = load_dataset("SetFit/ag_news", split="train")
data.features

settings = rg.Settings(
    fields=[rg.TextField(name="text")],
    questions=[
        rg.LabelQuestion(
            name="label", title="Classify the text:", labels=data.unique("label_text")
        ),
        rg.SpanQuestion(
            name="entities",
            title="Highlight all the entities in the text:",
            labels=["PERSON", "ORG", "LOC", "EVENT"],
            field="text",
        ),
    ],
)

dataset = rg.Dataset(name="ag_news", settings=settings)

dataset.create()

dataset.records.log(data, mapping={"label_text": "label"})