Spaces:
Sleeping
Sleeping
Andrew Green
commited on
Commit
·
0e4ad79
1
Parent(s):
a34dac3
Add text kwarg to help pipeline maybe?
Browse files
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']]
|