dataset kongo

app.py

@@ -13,9 +13,7 @@ from datasets import load_dataset
 
 ACTIONS = ["TRIP", "GITHUB", "MAIL", "CALENDAR"]
 NUM_ACTIONS = len(ACTIONS)
-DATASET_PATH = "iteratehack/code19-dataset"
-HF_SPLIT = "train"
-HF_DATASET = "iteratehack/code-19-dataset"
+HF_DATASET = "iteratehack/code19-dataset"
 
 # Confidence threshold - below this returns NONE
 CONFIDENCE_THRESHOLD = 0.6
@@ -233,11 +231,7 @@ class RLAgent:
 
 
 def load_hf_dataset():
-    dataset = load_dataset(
-        HF_DATASET,
-        data_files="dataset.jsonl",
-        split=HF_SPLIT
-    )
+    dataset = load_dataset(HF_DATASET, split="train")
     data = []
     for item in dataset:
         user_msg = item["messages"][1]["content"]