Update app.py

app.py

@@ -10,7 +10,7 @@ def load_and_preprocess_data():
 
     # Extract "prompt" and "chosen"/"rejected" as input-output pairs
     def process_row(example):
-        row = example['row']  # Access the 'row' key inside the example
+        row = example['rows']  # Access the 'row' key inside the example
         return {
             'prompt': row['prompt'],
             'chosen': row['chosen'],