wzkariampuzha commited on
Commit
8d4dc06
1 Parent(s): 801aede

Update classify_abs.py

Browse files
Files changed (1) hide show
  1. 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