fix: working example with argillalabeller updates

app.py

@@ -7,7 +7,7 @@ from distilabel.steps.tasks.argillalabeller import ArgillaLabeller
 llm = InferenceEndpointsLLM(
     model_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
     tokenizer_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
-    generation_kwargs={"max_new_tokens": 1000 * 128},
+    generation_kwargs={"max_new_tokens": 1000 * 4},
 )
 task = ArgillaLabeller(llm=llm)
 task.load()
@@ -30,7 +30,7 @@ def process_fields(fields):
     return [field if isinstance(field, dict) else json.loads(field) for field in fields]
 
 
-def process_records_gradio(records, example_records, fields, question):
+def process_records_gradio(records, fields, question, example_records=None):
     try:
         # Convert string inputs to dictionaries
         records = json.loads(records)
@@ -48,16 +48,15 @@ def process_records_gradio(records, example_records, fields, question):
         task.set_runtime_parameters(runtime_parameters)
 
         results = []
-        output = task.process(inputs=[{"records": record} for record in records])
+        output = task.process(inputs=[{"record": record} for record in records])
         for _ in range(len(records)):
             entry = next(output)[0]
             if entry["suggestions"]:
-                results.append(entry["suggestions"].serialize())
+                results.append(entry["suggestions"])
 
         return json.dumps({"results": results}, indent=2)
     except Exception as e:
         raise Exception(f"Error: {str(e)}")
-        return f"Error: {str(e)}"
 
 
 description = """