Commit
·
9edc4d0
1
Parent(s):
afd47cf
Update app.py
Browse files
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:
|