rbgo commited on
Commit
bb79aee
·
1 Parent(s): 9a4a3ed

loader added

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -7,13 +7,16 @@ import model
7
  import inference
8
 
9
 
10
- # Global params
11
- if 'model' not in st.session_state:
12
- loaded_model,tokenizer_eng,tokenizer_ass,in_input_length = model.main()
13
  st.session_state['model'] = loaded_model
14
  st.session_state['tokenizer_eng'] = tokenizer_eng
15
  st.session_state['tokenizer_ass'] = tokenizer_ass
16
  st.session_state['in_input_length'] = in_input_length
 
 
 
17
 
18
  # st.write(st.session_state)
19
 
 
7
  import inference
8
 
9
 
10
+ with st.spinner('Wait! Your Translator is loading...'):
11
+ if 'model' not in st.session_state:
12
+ loaded_model,tokenizer_eng,tokenizer_ass,in_input_length = model.main()
13
  st.session_state['model'] = loaded_model
14
  st.session_state['tokenizer_eng'] = tokenizer_eng
15
  st.session_state['tokenizer_ass'] = tokenizer_ass
16
  st.session_state['in_input_length'] = in_input_length
17
+ st.success('Done!')
18
+ # Global params
19
+
20
 
21
  # st.write(st.session_state)
22