Update app.py

app.py

@@ -3,8 +3,8 @@ import streamlit as st
 import torch
 import string
 
-
-
+bert_tokenizer = None
+bert_model = None
 
 from transformers import BertTokenizer, BertForMaskedLM
 
@@ -70,6 +70,8 @@ def get_bert_prediction(input_text,top_k):
  
 def run_test(sent,top_k):
   start = None
+  global bert_tokenizer
+  global bert_model
   if (bert_tokenizer is None):
         bert_tokenizer, bert_model  = load_bert_model(model_name)
   with st.spinner("Computing"):
@@ -104,8 +106,6 @@ print(top_k)
     
   #  with st.spinner("Computing"):
 try:
-      bert_tokenizer = None
-      bert_model = None
             
       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")
       option = st.selectbox(