Commit
·
9d7bba2
1
Parent(s):
2174a54
Update app.py
Browse files
app.py
CHANGED
@@ -76,6 +76,7 @@ def run_test(sent,top_k,model_name,display_area):
|
|
76 |
display_area.text("Loading model:" + st.session_state['model_name'])
|
77 |
st.session_state['bert_tokenizer'], st.session_state['bert_model'] = load_bert_model(st.session_state['model_name'])
|
78 |
try:
|
|
|
79 |
res = get_bert_prediction(sent,st.session_state['top_k'],st.session_state['model_name'])
|
80 |
return res
|
81 |
|
@@ -153,7 +154,7 @@ def main():
|
|
153 |
|
154 |
input_status_area = st.empty()
|
155 |
display_area = st.empty()
|
156 |
-
display_area.text("
|
157 |
if submit_button:
|
158 |
start = time.time()
|
159 |
if (len(text_input) == 0):
|
|
|
76 |
display_area.text("Loading model:" + st.session_state['model_name'])
|
77 |
st.session_state['bert_tokenizer'], st.session_state['bert_model'] = load_bert_model(st.session_state['model_name'])
|
78 |
try:
|
79 |
+
display_area.text("Model " + str(st.session_state['model_name']) + " load complete")
|
80 |
res = get_bert_prediction(sent,st.session_state['top_k'],st.session_state['model_name'])
|
81 |
return res
|
82 |
|
|
|
154 |
|
155 |
input_status_area = st.empty()
|
156 |
display_area = st.empty()
|
157 |
+
display_area.text("Current selections: results count = " + str(st.session_state['top_k']) + " ; " + " Model name = " + st.session_state['model_name'])
|
158 |
if submit_button:
|
159 |
start = time.time()
|
160 |
if (len(text_input) == 0):
|