File size: 975 Bytes
ddff90b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import pandas as pd
import classify_abs
#classify_abs is a dependency for extract_abs
import extract_abs
pd.set_option('display.max_colwidth', None)
import streamlit as st

#LSTM RNN Epi Classifier Model
classify_model_vars = classify_abs.init_classify_model()

#GARD Dictionary - For filtering and exact match disease/GARD ID identification
GARD_dict, max_length = extract_abs.load_GARD_diseases()

#BioBERT-based NER pipeline, open `entities` to see 
NER_pipeline, entity_classes = extract_abs.init_NER_pipeline()

#max_results is Maximum number of PubMed ID's to retrieve BEFORE filtering

#filtering options are 'strict','lenient'(default), 'none'
if text:
  out = extract_abs.search_term_extraction(term, max_results, filtering,
                                              NER_pipeline, entity_classes, 
                                              extract_diseases,GARD_dict, max_length, 
                                              classify_model_vars)
  st.(out)