Commit
·
41b19ce
1
Parent(s):
424d29e
Update app.py
Browse files
app.py
CHANGED
@@ -77,7 +77,7 @@ def get_bert_prediction(input_text,top_k,model_name):
|
|
77 |
def run_test(sent,top_k,model_name):
|
78 |
start = None
|
79 |
if (st.session_state['bert_tokenizer'] is None):
|
80 |
-
st.info("Loading model:"
|
81 |
st.session_state['bert_tokenizer'], st.session_state['bert_model'] = load_bert_model(st.session_state['model_name'])
|
82 |
with st.spinner("Computing"):
|
83 |
start = time.time()
|
|
|
77 |
def run_test(sent,top_k,model_name):
|
78 |
start = None
|
79 |
if (st.session_state['bert_tokenizer'] is None):
|
80 |
+
st.info("Loading model:" + st.session_state['model_name'])
|
81 |
st.session_state['bert_tokenizer'], st.session_state['bert_model'] = load_bert_model(st.session_state['model_name'])
|
82 |
with st.spinner("Computing"):
|
83 |
start = time.time()
|