update app
Browse files
app.py
CHANGED
@@ -17,13 +17,12 @@ def load_model(model_ckpt):
|
|
17 |
|
18 |
@st.cache()
|
19 |
def load_examples():
|
20 |
-
with open("examples.json", "r") as f:
|
21 |
examples = json.load(f)
|
22 |
return examples
|
23 |
|
24 |
st.set_page_config(page_icon=":laptop:", layout="wide")
|
25 |
|
26 |
-
|
27 |
st.sidebar.header("Models")
|
28 |
models = ["CodeParrot", "InCoder"]
|
29 |
selected_models = st.sidebar.multiselect("Select code generation models to compare", models, default=["CodeParrot"])
|
@@ -35,14 +34,14 @@ selected_task = st.sidebar.selectbox("Select a task", tasks)
|
|
35 |
|
36 |
if selected_task == " ":
|
37 |
st.title("Code Generation Models")
|
38 |
-
with open("intro.txt", "r") as f:
|
39 |
intro = f.read()
|
40 |
st.markdown(intro)
|
41 |
|
42 |
elif selected_task == "Pretraining datasets":
|
43 |
st.title("Pretraining datasets π")
|
44 |
st.markdown("Preview of some code files from Github repositories")
|
45 |
-
df = pd.read_csv("data_preview.csv")
|
46 |
st.dataframe(df)
|
47 |
for model in selected_models:
|
48 |
with open(f"datasets/{model.lower()}.txt", "r") as f:
|
@@ -72,7 +71,7 @@ elif selected_task == "Code generation":
|
|
72 |
examples = load_examples()
|
73 |
example_names = [example["name"] for example in examples]
|
74 |
name2id = dict([(name, i) for i, name in enumerate(example_names)])
|
75 |
-
selected_example = st.sidebar.selectbox("Select one of the following examples", example_names)
|
76 |
example_text = examples[name2id[selected_example]]["value"]
|
77 |
default_length = examples[name2id[selected_example]]["length"]
|
78 |
st.sidebar.header("Generation settings")
|
|
|
17 |
|
18 |
@st.cache()
|
19 |
def load_examples():
|
20 |
+
with open("utils/examples.json", "r") as f:
|
21 |
examples = json.load(f)
|
22 |
return examples
|
23 |
|
24 |
st.set_page_config(page_icon=":laptop:", layout="wide")
|
25 |
|
|
|
26 |
st.sidebar.header("Models")
|
27 |
models = ["CodeParrot", "InCoder"]
|
28 |
selected_models = st.sidebar.multiselect("Select code generation models to compare", models, default=["CodeParrot"])
|
|
|
34 |
|
35 |
if selected_task == " ":
|
36 |
st.title("Code Generation Models")
|
37 |
+
with open("utils/intro.txt", "r") as f:
|
38 |
intro = f.read()
|
39 |
st.markdown(intro)
|
40 |
|
41 |
elif selected_task == "Pretraining datasets":
|
42 |
st.title("Pretraining datasets π")
|
43 |
st.markdown("Preview of some code files from Github repositories")
|
44 |
+
df = pd.read_csv("utils/data_preview.csv")
|
45 |
st.dataframe(df)
|
46 |
for model in selected_models:
|
47 |
with open(f"datasets/{model.lower()}.txt", "r") as f:
|
|
|
71 |
examples = load_examples()
|
72 |
example_names = [example["name"] for example in examples]
|
73 |
name2id = dict([(name, i) for i, name in enumerate(example_names)])
|
74 |
+
selected_example = st.sidebar.selectbox("Select one of the following examples or implement yours", example_names)
|
75 |
example_text = examples[name2id[selected_example]]["value"]
|
76 |
default_length = examples[name2id[selected_example]]["length"]
|
77 |
st.sidebar.header("Generation settings")
|