Update app.py

app.py

@@ -131,10 +131,13 @@ model_id = 0 if model_choice == MODEL_OPTIONS[0] else 1
 
 if 'query' not in st.session_state:
     query = st.text_input("", value="artificial intelligence", key="query")
-    #st.session_state['query'] = 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