new datasets

app.py

@@ -2,19 +2,24 @@ import gradio as gr
 import argilla as rg
 import os
 
-from argilla._constants import DEFAULT_API_KEY
 from datasets import load_dataset
 
 dataset = load_dataset("dvilasuero/banking_app", split="train").shuffle()
-my_dataset = load_dataset("argilla/news", split="test")
+# You can find your Space URL behind the Embed this space button
+# Change it
+rg.init(
+    api_url="https://ravi259-sml-argilla.hf.space", 
+    api_key="admin.apikey"
+)
+banking_ds = load_dataset("argilla/banking_sentiment_setfit", split="train")
+# Argilla expects labels in the annotation column
+# We include labels for demo purposes
+banking_ds = banking_ds.rename_column("label", "annotation")
+# Build argilla dataset from datasets
+argilla_ds = rg.read_datasets(banking_ds, task="TextClassification")
 
-# import data from a datasets Dataset
-dataset_rg = rg.read_datasets(my_dataset, task="TextClassification")
-# or
-dataset_rg = rg.DatasetForTextClassification.from_datasets(my_dataset)
-
-# export back to a datasets
-dataset_rg.to_datasets()
+# Create dataset
+rg.log(argilla_ds, "bankingapp_sentiment")
 
 def greet(name):
     return "Hello " + name + "!!"