ajitrajasekharan commited on
Commit
0a7c967
·
1 Parent(s): 03be735

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ start = None
71
  # with st.spinner("Computing"):
72
  try:
73
 
74
- 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'], index=0, key = "model_name")
75
  bert_tokenizer, bert_model = load_bert_model(model_name)
76
  default_text = "Imatinib is used to treat"
77
  input_text = st.text_area(
 
71
  # with st.spinner("Computing"):
72
  try:
73
 
74
+ 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")
75
  bert_tokenizer, bert_model = load_bert_model(model_name)
76
  default_text = "Imatinib is used to treat"
77
  input_text = st.text_area(