Update app.py

app.py

@@ -95,6 +95,7 @@ if st.button("🎲 Pick Random Question"):
 
 default_prompt = "Jasper has 5 apples and eats 2 of them. How many apples does he have left?"
 selected_question = gsm8k_data[question_index]["question"] if question_index is not None else default_prompt
+correct_answer = gsm8k_data[question_index]["answer"]
 
 
 # Prompt options
@@ -137,7 +138,10 @@ if st.button("Generate Response", key="manual"):
                 generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
                 response_only = generated_text[len(input_text):].strip() if generated_text.startswith(input_text) else generated_text.strip()
 
-    st.subheader("🔎 Prompt")
-    st.write(input_text)
+    with st.expander("🔎 Prompt"):
+        st.subheader("🔎 Prompt")
+        st.write(input_text)
     st.subheader("🧠 Model Output")
     st.success(response_only)
+    st.subheader("✅ Correct Answer (GSM8K)")
+    st.info(correct_answer)