Spaces:
Runtime error
Runtime error
Commit
·
8d4dc06
1
Parent(s):
801aede
Update classify_abs.py
Browse files- classify_abs.py +1 -11
classify_abs.py
CHANGED
|
@@ -423,14 +423,4 @@ def getTextPredictions(abstract:str, classify_model_vars:Tuple[Any,Any,Any,Any,A
|
|
| 423 |
return prob, isEpi
|
| 424 |
|
| 425 |
else:
|
| 426 |
-
return 0.0, False
|
| 427 |
-
|
| 428 |
-
if __name__ == '__main__':
|
| 429 |
-
print('Loading 5 NLP models...')
|
| 430 |
-
classify_model_vars= init_classify_model()
|
| 431 |
-
print('All models loaded.')
|
| 432 |
-
pmid = input('\nEnter PubMed PMID (or DONE): ')
|
| 433 |
-
while pmid != 'DONE':
|
| 434 |
-
abstract, prob, isEpi = getPredictions(pmid, classify_model_vars)
|
| 435 |
-
print(abstract, prob, isEpi)
|
| 436 |
-
pmid = input('\nEnter PubMed PMID (or DONE): ')
|
|
|
|
| 423 |
return prob, isEpi
|
| 424 |
|
| 425 |
else:
|
| 426 |
+
return 0.0, False
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|