skaltenp commited on
Commit
c5380cb
·
verified ·
1 Parent(s): 94cbf8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -41,9 +41,8 @@ def prepare_sample_text(example, tokenizer, remove_indent=False, start=None, end
41
  return text
42
 
43
  dataset = load_dataset(
44
- args.dataset_name,
45
  token=True,
46
- num_proc=args.num_workers,
47
  download_mode='force_redownload'
48
  )
49
  train_data = dataset["train"].train_test_split(train_size=0.8, shuffle=True, seed=199900595)
 
41
  return text
42
 
43
  dataset = load_dataset(
44
+ "skaltenp/sepsis_cases",
45
  token=True,
 
46
  download_mode='force_redownload'
47
  )
48
  train_data = dataset["train"].train_test_split(train_size=0.8, shuffle=True, seed=199900595)