Update app.py

app.py

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