ajitrajasekharan commited on
Commit
9edc4d0
·
1 Parent(s): afd47cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -80,13 +80,16 @@ start = None
80
  try:
81
 
82
  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")
 
 
 
83
  bert_tokenizer, bert_model = load_bert_model(model_name)
84
  default_text = "Imatinib is used to [MASK] nsclc"
85
  input_text = st.text_area(
86
  label="Enter text below",
87
  value=default_text,
88
  )
89
- if st.button("Submit"):
90
  with st.spinner("Computing"):
91
  start = time.time()
92
  try:
 
80
  try:
81
 
82
  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")
83
+ option = st.selectbox(
84
+ 'Choose any of these sentences to test models',
85
+ ('', "[MASK] who lives in New York and works for XCorp suffers from Parkinson's", "Lou Gehrig who lives in [MASK] and works for XCorp suffers from Parkinson's","'Lou Gehrig who lives in New York and works for [MASK] suffers from Parkinson's'","'Lou Gehrig who lives in New York and works for XCorp suffers from [MASK]'"))
86
  bert_tokenizer, bert_model = load_bert_model(model_name)
87
  default_text = "Imatinib is used to [MASK] nsclc"
88
  input_text = st.text_area(
89
  label="Enter text below",
90
  value=default_text,
91
  )
92
+ if st.button("Submit") or len(option) > 0:
93
  with st.spinner("Computing"):
94
  start = time.time()
95
  try: