Update

app.py

@@ -104,14 +104,15 @@ else:
     with tab_submission:
         st.header("Submissions")
         models = get_dataset_config_names("Elfsong/Venus_Model_Evaluation")
-        option = st.selectbox("Which model you are looking for?", models, placeholder="Select a model...")
-        st.write("You selected:", option)
+        model_name = st.selectbox("Which model you are looking for?", models, placeholder="Select a model...")
+        st.write("You selected:", model_name)
 
         with st.spinner("Loading Data...", show_time=True):
+            ds = load_dataset("Elfsong/Venus_Model_Evaluation", model_name, split='train')
             df = pd.DataFrame(
                 {
-                    "problem_id": [int(problem['problem_id']) for problem in problem_dict.values()],
-                    "solution": [str(problem['solution']) for problem in problem_dict.values()],
+                    "problem_id": [int(problem['problem_id']) for problem in ds],
+                    "solution": [str(problem['solution']) for problem in ds],
                 }
             )
         st.dataframe(