Update src/streamlit_app.py

src/streamlit_app.py

@@ -82,7 +82,8 @@ with tab2:
     st.markdown("A random question from GSM8K will be shown. Click below to test the model.")
 
     if st.button("Run GSM8K Sample"):
-        gsm8k_path = "data/gsm8k"
+        ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+        gsm8k_path = os.path.join(ROOT_DIR, "data", "gsm8k")
         files = [f for f in os.listdir(gsm8k_path) if f.endswith(".jsonl")]
         if not files:
             st.error("No JSON files found in data/gsm8k.")