my-argilla / argilla.py
Ihssane123's picture
Upload argilla.py
ef915aa verified
raw
history blame contribute delete
991 Bytes
import argilla as rg
from datasets import load_dataset
client = rg.Argilla(
api_url="https://ihssane123-my-argilla.hf.space",
api_key="UJ-3ANbta5mlYiUgJJxWPtp7nr4mXd6wvXEgl2AVi4uEk1ChOdLAjtu1n1lNSYJEB2WylAFRXGvJe6m4cynzk_U4o1twaBP6Kapk85XZIZY",
)
settings = rg.Settings(
guidelines="Classify the reviews as positive or negative.",
fields=[
rg.TextField(
name="review",
title="Text from the review",
use_markdown=False,
),
],
questions=[
rg.LabelQuestion(
name="my_label",
title="In which category does this article fit?",
labels=["positive", "negative"],
)
],
)
dataset = rg.Dataset(
name=f"my_first_dataset",
settings=settings,
client=client,
#workspace="argilla"
)
dataset.create()
data = load_dataset("imdb", split="train[:100]").to_list()
dataset.records.log(records=data, mapping={"text": "review"})