fix: update example

app.py

@@ -50,7 +50,7 @@ def load_examples():
 
 
 # Create Gradio examples
-examples = load_examples()
+examples = load_examples()[:1]
 
 
 def process_fields(fields):
@@ -112,34 +112,44 @@ from gradio_client import Client
 import argilla as rg
 
 # Initialize Argilla client
-client = rg.Argilla(
+gradio_client = Client("davidberenstein1957/distilabel-argilla-labeller")
+argilla_client = rg.Argilla(
     api_key=os.environ["ARGILLA_API_KEY"], api_url=os.environ["ARGILLA_API_URL"]
 )
 
 # Load the dataset
-dataset = client.datasets(name="my_dataset", workspace="my_workspace")
-
-# Prepare example data
-example_field = dataset.settings.fields["my_input_field"].serialize()
-example_question = dataset.settings.questions["my_question_to_predict"].serialize()
+dataset = argilla_client.datasets(name="my_dataset", workspace="my_workspace")
+
+# Get the field and question
+field = dataset.settings.fields["text"]
+question = dataset.settings.questions["sentiment"]
+
+# Get completed and pending records
+completed_records_filter = rg.Filter(("status", "==", "completed"))
+pending_records_filter = rg.Filter(("status", "==", "pending"))
+example_records = list(
+    dataset.records(
+        query=rg.Query(filter=completed_records_filter),
+        limit=5,
+    )
+)
+some_pending_records = list(
+    dataset.records(
+        query=rg.Query(filter=pending_records_filter),
+        limit=5,
+    )
+)
 
+# Process the records
 payload = {
-    "records": [next(dataset.records()).to_dict()],
-    "fields": [example_field],
-    "question": example_question,
+    "records": [record.to_dict() for record in some_pending_records],
+    "fields": [field.serialize()],
+    "question": question.serialize(),
+    "example_records": [record.to_dict() for record in example_records],
+    "api_name": "/predict",
 }
 
-# Use gradio client to process the data
-client = Client("davidberenstein1957/distilabel-argilla-labeller")
-
-result = client.predict(
-    records=json.dumps(payload["records"]),
-    example_records=json.dumps(payload["example_records"]),
-    fields=json.dumps(payload["fields"]),
-    question=json.dumps(payload["question"]),
-    api_name="/predict"
-)
-
+response = gradio_client.predict(**payload)
 ```
 """