ajitrajasekharan
commited on
Commit
•
07c19db
1
Parent(s):
d9f055f
Update app.py
Browse files
app.py
CHANGED
@@ -72,6 +72,7 @@ def get_bert_prediction(input_text,top_k,model_name):
|
|
72 |
def run_test(sent,top_k,model_name):
|
73 |
start = None
|
74 |
if (st.session_state['bert_tokenizer'] is None):
|
|
|
75 |
st.session_state['bert_tokenizer'], st.session_state['bert_model'] = load_bert_model(st.session_state['model_name'])
|
76 |
with st.spinner("Computing"):
|
77 |
start = time.time()
|
@@ -94,6 +95,7 @@ def on_text_change():
|
|
94 |
def on_option_change():
|
95 |
|
96 |
text = st.session_state.my_choice
|
|
|
97 |
run_test(text,st.session_state['top_k'],st.session_state['model_name'])
|
98 |
|
99 |
def on_results_count_change():
|
|
|
72 |
def run_test(sent,top_k,model_name):
|
73 |
start = None
|
74 |
if (st.session_state['bert_tokenizer'] is None):
|
75 |
+
st.info("Loading model:",st.session_state['model_name'])
|
76 |
st.session_state['bert_tokenizer'], st.session_state['bert_model'] = load_bert_model(st.session_state['model_name'])
|
77 |
with st.spinner("Computing"):
|
78 |
start = time.time()
|
|
|
95 |
def on_option_change():
|
96 |
|
97 |
text = st.session_state.my_choice
|
98 |
+
st.info("Preselected text chosen")
|
99 |
run_test(text,st.session_state['top_k'],st.session_state['model_name'])
|
100 |
|
101 |
def on_results_count_change():
|