Andrew Green commited on
Commit
0e4ad79
·
1 Parent(s): a34dac3

Add text kwarg to help pipeline maybe?

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def get_pipeline():
34
  def classify_abstracts(abstracts:Dict[str, str]) -> None:
35
  pipe = get_pipeline()
36
  pmids = list(abstracts.keys())
37
- classification = pipe(list(abstracts.values()))
38
 
39
  for pmid, abs in zip(pmids, classification):
40
  abs['label'] = label_lookup[abs['label']]
 
34
  def classify_abstracts(abstracts:Dict[str, str]) -> None:
35
  pipe = get_pipeline()
36
  pmids = list(abstracts.keys())
37
+ classification = pipe(text=list(abstracts.values()))
38
 
39
  for pmid, abs in zip(pmids, classification):
40
  abs['label'] = label_lookup[abs['label']]