Update app.py

app.py

@@ -117,8 +117,9 @@ with st.sidebar:
         submitted = st.form_submit_button("Start Computing")
 
 
-if submitted:
 
+if submitted or ("submitted" in st.session_state):
+    st.session_state["submitted"] = True
     config = model_configs[model_name]
 
     config.inference_sample_num = molecule_num_input