Update pages/type_text.py

pages/type_text.py

@@ -175,7 +175,7 @@ if INTdesc_input and st.button(":blue[Map to SBS codes]", key="run_st_model"): #
     #st.markdown('<div id="bottom"></div>', unsafe_allow_html=True)
     #components.html(scroll_script, height=0, width=0)
 
-    st.subheader("Reasoning Model Output:") # optional Add a subheader for clarity
+    
     ## Define the Reasoning models
     rs_models = {
         '(medium speed) original model for general domain: meta-llama/Llama-3.2-1B-Instruct': 'meta-llama/Llama-3.2-1B-Instruct',
@@ -218,6 +218,7 @@ if INTdesc_input and st.button(":blue[Map to SBS codes]", key="run_st_model"): #
             messages,
             max_new_tokens=256,
         )
+    st.subheader("Reasoning Model Output:") # optional Add a subheader for clarity
     st.write(outputs[0]["generated_text"][-1]["content"])
 
     # Download buttons