Spaces:
Running
Running
wzkariampuzha
commited on
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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|