Loader updated

app.py

@@ -7,14 +7,14 @@ import model
 import inference
 
 
-with st.spinner('Wait! Your Translator is loading...'):
+with st.spinner('Wait for it...'):
     if 'model' not in st.session_state:
         loaded_model,tokenizer_eng,tokenizer_ass,in_input_length = model.main()
-    st.session_state['model'] = loaded_model
-    st.session_state['tokenizer_eng'] = tokenizer_eng
-    st.session_state['tokenizer_ass'] = tokenizer_ass
-    st.session_state['in_input_length'] = in_input_length
-st.success('Done!')
+        st.session_state['model'] = loaded_model
+        st.session_state['tokenizer_eng'] = tokenizer_eng
+        st.session_state['tokenizer_ass'] = tokenizer_ass
+        st.session_state['in_input_length'] = in_input_length
+# st.success('Done!')
 # Global params