Update app.py

app.py

@@ -17,7 +17,7 @@ def load_data(lang):
         split = "lean_proofsteps"
     elif lang == "Isabelle Proofsteps":
         split = "isa_proofsteps"
-    ds = load_dataset("xu3kev/proof-pile-2-proofsteps", split="lean_proofsteps[:2%]")
+    ds = load_dataset("xu3kev/proof-pile-2-proofsteps", split=f"{split}[:2%]")
     return ds
 
 list_languages = ['Lean Proofsteps', 'Isabelle Proofsteps']