ajitrajasekharan commited on
Commit
7c8c08d
·
1 Parent(s): 167f69d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -4,8 +4,7 @@ import torch
4
  import string
5
 
6
 
7
- bert_tokenizer = None
8
- bert_model = None
9
 
10
  from transformers import BertTokenizer, BertForMaskedLM
11
 
@@ -105,6 +104,8 @@ print(top_k)
105
 
106
  # with st.spinner("Computing"):
107
  try:
 
 
108
 
109
  model_name = st.sidebar.selectbox(label='Select Model to Apply', options=['ajitrajasekharan/biomedical', 'bert-base-cased','bert-large-cased','microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext','allenai/scibert_scivocab_cased'], index=0, key = "model_name")
110
  option = st.selectbox(
 
4
  import string
5
 
6
 
7
+
 
8
 
9
  from transformers import BertTokenizer, BertForMaskedLM
10
 
 
104
 
105
  # with st.spinner("Computing"):
106
  try:
107
+ bert_tokenizer = None
108
+ bert_model = None
109
 
110
  model_name = st.sidebar.selectbox(label='Select Model to Apply', options=['ajitrajasekharan/biomedical', 'bert-base-cased','bert-large-cased','microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext','allenai/scibert_scivocab_cased'], index=0, key = "model_name")
111
  option = st.selectbox(