Update app.py

app.py

@@ -17,6 +17,15 @@ DESCRIPTION = """
 Built with 🤗 Hugging Face's [transformers](https://huggingface.co/transformers/) library, [SentenceBert](https://www.sbert.net/) models, [Streamlit](https://streamlit.io/) and 44k movie descriptions from the Kaggle [Movies Dataset](https://www.kaggle.com/rounakbanik/the-movies-dataset)
 """
 
+if 'query' not in st.session_state:
+    query = st.text_input("", value="artificial intelligence", key="query")
+    st.session_state['query'] = 'value'
+else:
+    query = st.text_input("", value=st.session_state["query"], key="query")
+    st.session_state.query = query 
+if 'query' not in st.session_state:
+    st.session_state.query = 'value'    
+st.write(st.session_state.query)
 
 @st.cache(
     show_spinner=False,
@@ -124,22 +133,6 @@ model_choice = st.sidebar.selectbox("Similarity model", options=MODEL_OPTIONS)
 model_id = 0 if model_choice == MODEL_OPTIONS[0] else 1
 
 
-
-#def form_callback():
-#    st.write(st.session_state.my_slider)
-#    st.write(st.session_state.my_checkbox)
-
-if 'query' not in st.session_state:
-    query = st.text_input("", value="artificial intelligence", key="query")
-    st.session_state['query'] = 'value'
-else:
-    query = st.text_input("", value=st.session_state["query"], key="query")
-    st.session_state.query = query 
-
-if 'query' not in st.session_state:
-    st.session_state.query = 'value'    
-st.write(st.session_state.query)
-
 # Session state
 if 'key' not in st.session_state:
     st.session_state['key'] = 'value'