Update app.py

app.py

@@ -60,7 +60,12 @@ def generate_code_threads(
     for t in threads:
         t.join()
 
-
+@st.cache()
+def generate_teaser(gen_prompt):
+    generations = []
+    generate_code(generations, "CodeGen", gen_prompt, 10, 0.2, 42)
+    st.code(generations[0]) 
+    
 st.set_page_config(page_icon=":laptop:", layout="wide")
 with open("utils/table_contents.md", "r") as f:
     contents = f.read()
@@ -79,12 +84,11 @@ with col2:
         value=example_text,
         height=100,
     ).strip()
+  
     if st.button("Generate code!", key=1):
         with st.spinner("Generating code..."):
-            # use threading
-            generations = []
-            generate_code(generations, "CodeGen", gen_prompt, 10, 0.2, 42)
-            st.code(generations[0])  
+            generate_teaser(gen_prompt)
+ 
 read_markdown("utils/intro.md")
 
 # Code datasets