paulbricman commited on
Commit
ac645f4
1 Parent(s): 240d95b

feat: add spinner to indicate loading

Browse files
Files changed (1) hide show
  1. main.py +6 -5
main.py CHANGED
@@ -104,11 +104,12 @@ else:
104
  raw_pars = st.session_state['content'].split('\n')
105
  pars = []
106
 
107
- for raw_par in raw_pars:
108
- if raw_par.strip() != '':
109
- corpus_tokens, attention = attend(
110
- raw_par, query, model, tokenizer)
111
- pars += [render_html(corpus_tokens, attention, focus)]
 
112
 
113
  if st.sidebar.button('reset content'):
114
  st.session_state['content'] = None
104
  raw_pars = st.session_state['content'].split('\n')
105
  pars = []
106
 
107
+ with st.spinner('attending to text...'):
108
+ for raw_par in raw_pars:
109
+ if raw_par.strip() != '':
110
+ corpus_tokens, attention = attend(
111
+ raw_par, query, model, tokenizer)
112
+ pars += [render_html(corpus_tokens, attention, focus)]
113
 
114
  if st.sidebar.button('reset content'):
115
  st.session_state['content'] = None