Elfsong commited on
Commit
b64b933
·
1 Parent(s): 554d373
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -11,24 +11,25 @@ from datasets import load_dataset
11
 
12
  st.title("Code:blue[Arena]")
13
 
14
- with st.spinner("Loading data...", show_time=True):
15
- problem_dict = dict()
16
 
17
- # Venus Data
 
18
  venus_ds = load_dataset("Elfsong/leetcode_data", split='train')
19
  for problem in venus_ds:
20
  problem_id = problem["title"]
21
  problem['type'] = "leetcode"
22
  problem_dict[problem_id] = problem
23
 
24
- # APPS Data
 
25
  apps_ds = load_dataset("Elfsong/APPS_Python", split='test')
26
  for problem in apps_ds:
27
  problem_id = f'apps_{problem["problem_id"]}'
28
  problem['type'] = "apps"
29
  problem_dict[problem_id] = problem
30
 
31
- problem_count = len(problem_dict)
32
 
33
 
34
  if "problem" in st.query_params:
 
11
 
12
  st.title("Code:blue[Arena]")
13
 
14
+ problem_dict = dict()
 
15
 
16
+ # Venus Data
17
+ with st.spinner("Loading Venus data...", show_time=True):
18
  venus_ds = load_dataset("Elfsong/leetcode_data", split='train')
19
  for problem in venus_ds:
20
  problem_id = problem["title"]
21
  problem['type'] = "leetcode"
22
  problem_dict[problem_id] = problem
23
 
24
+ # APPS Data
25
+ with st.spinner("Loading APPS data...", show_time=True):
26
  apps_ds = load_dataset("Elfsong/APPS_Python", split='test')
27
  for problem in apps_ds:
28
  problem_id = f'apps_{problem["problem_id"]}'
29
  problem['type'] = "apps"
30
  problem_dict[problem_id] = problem
31
 
32
+ problem_count = len(problem_dict)
33
 
34
 
35
  if "problem" in st.query_params: