Elfsong commited on
Commit
87f8f5e
·
1 Parent(s): feb488c
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -17,6 +17,7 @@ with st.spinner("Loading data...", show_time=True):
17
  for problem in ds:
18
  problem_id = problem["problem_id"]
19
  problem_dict[problem_id] = problem
 
20
 
21
 
22
  if "problem" in st.query_params:
@@ -78,14 +79,16 @@ else:
78
  df = pd.DataFrame(
79
  {
80
  "model_name": [model_name for model_name in model_list],
 
81
  }
82
  )
83
 
84
  st.dataframe(
85
  df,
86
  column_config={
87
- "model_name": st.column_config.TextColumn("Problem ID"),
 
88
  },
89
- column_order=("model_name",),
90
  hide_index=True,
91
  )
 
17
  for problem in ds:
18
  problem_id = problem["problem_id"]
19
  problem_dict[problem_id] = problem
20
+ problem_count = len(problem_dict)
21
 
22
 
23
  if "problem" in st.query_params:
 
79
  df = pd.DataFrame(
80
  {
81
  "model_name": [model_name for model_name in model_list],
82
+ "model_progress": [random.randint(0, problem_count+1) for model_name in model_list],
83
  }
84
  )
85
 
86
  st.dataframe(
87
  df,
88
  column_config={
89
+ "model_name": st.column_config.TextColumn("Model Name"),
90
+ "model_progress": st.column_config.ProgressColumn("Progress", min_value=0, max_value=problem_count),
91
  },
92
+ column_order=("model_name", "model_progress"),
93
  hide_index=True,
94
  )