Benjamin S Liang commited on
Commit
50da02d
1 Parent(s): 86bb186

Fix dataset loading

Browse files
app.py CHANGED
@@ -21,8 +21,9 @@ def load_abstracts():
21
  val_filing_start_date='2016-01-01',
22
  val_filing_end_date='2016-01-01',
23
  )
24
- dataset_dict = [] # free up space
25
- return dataset_dict['train']['abstract']
 
26
 
27
  # returns loaded model and tokenizer, if any
28
  def load_model(opt):
 
21
  val_filing_start_date='2016-01-01',
22
  val_filing_end_date='2016-01-01',
23
  )
24
+ abstracts = dataset_dict['train']['abstract']
25
+ dataset_dict = []
26
+ return abstracts
27
 
28
  # returns loaded model and tokenizer, if any
29
  def load_model(opt):
finetunehupd.ipynb CHANGED
@@ -1281,6 +1281,14 @@
1281
  "dataset_dict['train']['abstract'][0]"
1282
  ]
1283
  },
 
 
 
 
 
 
 
 
1284
  {
1285
  "cell_type": "code",
1286
  "execution_count": 6,
 
1281
  "dataset_dict['train']['abstract'][0]"
1282
  ]
1283
  },
1284
+ {
1285
+ "cell_type": "code",
1286
+ "execution_count": null,
1287
+ "id": "cd6f20fc-9874-465d-9781-505415db3ffd",
1288
+ "metadata": {},
1289
+ "outputs": [],
1290
+ "source": []
1291
+ },
1292
  {
1293
  "cell_type": "code",
1294
  "execution_count": 6,
images/machine_learning_banner.jpg ADDED