Loubna ben allal commited on
Commit
7982bc6
β€’
1 Parent(s): 5b9eb09

update app

Browse files
Files changed (1) hide show
  1. app.py +11 -10
app.py CHANGED
@@ -21,27 +21,28 @@ def load_examples():
21
  return examples
22
 
23
  st.set_page_config(page_icon=':laptop:', layout="wide")
24
-
25
- st.title("Code Generation Models comparison πŸ’»")
26
- with open("intro.txt", "r") as f:
27
- intro = f.read()
28
- st.markdown(intro)
29
 
30
 
31
- st.sidebar.header("Models:")
32
  models = ["CodeParrot", "OPT", "InCoder"]
33
- selected_models = st.sidebar.multiselect('Select code generation models to compare',
34
  models,
35
  default=["CodeParrot"])
36
- st.sidebar.header("Tasks:")
37
- tasks = ["Model architecture", "Model evaluation", "Pretraining dataset", "Prompting"]
38
  selected_task = st.sidebar.selectbox("Select a task:", tasks)
39
 
40
  architectures = {}
41
  datasets = {}
42
  pipelines = {}
43
 
44
- if selected_task == "Pretraining dataset":
 
 
 
 
 
 
45
  st.title("Pretraining datasets πŸ“š")
46
  for model in selected_models:
47
  with open(f"datasets/{model.lower()}.txt", "r") as f:
 
21
  return examples
22
 
23
  st.set_page_config(page_icon=':laptop:', layout="wide")
 
 
 
 
 
24
 
25
 
26
+ st.sidebar.header("Models")
27
  models = ["CodeParrot", "OPT", "InCoder"]
28
+ selected_models = st.sidebar.multiselect('Select code generation models to compare:',
29
  models,
30
  default=["CodeParrot"])
31
+ st.sidebar.header("Tasks")
32
+ tasks = [" ","Model architecture", "Model evaluation", "Pretraining dataset", "Prompting"]
33
  selected_task = st.sidebar.selectbox("Select a task:", tasks)
34
 
35
  architectures = {}
36
  datasets = {}
37
  pipelines = {}
38
 
39
+ if sselected_task = " ":
40
+ st.title("Code Generation Models comparison πŸ’»")
41
+ with open("intro.txt", "r") as f:
42
+ intro = f.read()
43
+ st.markdown(intro)
44
+
45
+ elif selected_task == "Pretraining dataset":
46
  st.title("Pretraining datasets πŸ“š")
47
  for model in selected_models:
48
  with open(f"datasets/{model.lower()}.txt", "r") as f: