ashmib commited on
Commit
ca0246c
·
verified ·
1 Parent(s): 0d4373e

Update utils/loaders.py

Browse files
Files changed (1) hide show
  1. utils/loaders.py +1 -1
utils/loaders.py CHANGED
@@ -22,7 +22,7 @@ def load_data():
22
  dataset = load_dataset(REPO_NAME, token=True, data_files=DATA_FILES, revision="main")
23
  dataset.set_format(type='pandas') ## converting it into pandas
24
  df = dataset["train"][:]
25
- return df
26
 
27
 
28
  def load_html(file_name):
 
22
  dataset = load_dataset(REPO_NAME, token=True, data_files=DATA_FILES, revision="main")
23
  dataset.set_format(type='pandas') ## converting it into pandas
24
  df = dataset["train"][:]
25
+ return df[:60]
26
 
27
 
28
  def load_html(file_name):