on_form_submit is now cached

hanlde_form_submit.py

@@ -1,3 +1,4 @@
+import streamlit as st
 from grouped_sampling import GroupedSamplingPipeLine
 
 
@@ -15,6 +16,7 @@ def create_pipeline(model_name: str, group_size) -> GroupedSamplingPipeLine:
     )
 
 
+@st.cache
 def on_form_submit(model_name: str, group_size: int, prompt: str) -> str:
     """
     Called when the user submits the form.