Update app.py

app.py

@@ -12,12 +12,20 @@ Here you can inspect proofsteps from [Proof-Pile-V2](https://huggingface.co/data
 """)
 
 @st.cache()
-def load_data():
-    ds = load_dataset("xu3kev/proof-pile-2-proofsteps", split="lean_proofsteps")
+def load_data(lang):
+    if lang == "Lean Proofsteps":
+        split = "lean_proofsteps"
+    elif lang == "Isabelle Proofsteps":
+        split = "isa_proofsteps"
+    ds = load_dataset("xu3kev/proof-pile-2-proofsteps", split="lean_proofsteps[:2%]")
     return ds
 
+list_languages = ['Lean Proofsteps', 'Isabelle Proofsteps']
+chosen_language = st.sidebar.selectbox(
+    label="Select a Proof Language", options=list_languages, index=0
+)
 
-samples = load_data()
+samples = load_data(chosen_language)
 st.sidebar.header('Sample Selection')
 index_example = st.sidebar.number_input(f"Choose a sample from the existing {len(samples)} notebooks:", min_value=0, max_value=max(0, len(samples)-1), value=0, step=1)