Add new column names to TARGET_COLUMN_NAMES

app.py

@@ -40,6 +40,8 @@ TARGET_COLUMN_NAMES = {
     "response",
     "context",
     "query",
+    "chosen",
+    "rejected",
 }