Jabuszko commited on
Commit
cd10cff
·
verified ·
1 Parent(s): 6dd93cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -233,8 +233,11 @@ class RLAgent:
233
 
234
 
235
  def load_hf_dataset():
236
- dataset = load_dataset(HF_DATASET, split=HF_SPLIT)
237
-
 
 
 
238
  data = []
239
  for item in dataset:
240
  user_msg = item["messages"][1]["content"]
 
233
 
234
 
235
  def load_hf_dataset():
236
+ dataset = load_dataset(
237
+ HF_DATASET,
238
+ data_files="dataset.jsonl",
239
+ split=HF_SPLIT
240
+ )
241
  data = []
242
  for item in dataset:
243
  user_msg = item["messages"][1]["content"]