update

app.py

@@ -30,7 +30,7 @@ def generate_code(model_name, gen_prompt, max_new_tokens, temperature, seed):
     return generated_text
 
 
-st.set_page_config(page_icon=":laptop:", layout="wide")
+#st.set_page_config(page_icon=":laptop:", layout="wide")
 
 # Introduction
 st.title("Code generation with 🤗")
@@ -46,10 +46,10 @@ st.markdown(
 df = pd.read_csv("utils/data_preview.csv")
 st.dataframe(df)
 st.header("Model")
-selected_model1 = st.selectbox(
+selected_model = st.selectbox(
     "Select a code generation model", MODELS, key=1
 )
-with open(f"datasets/{selected_model1.lower()}.txt", "r") as f:
+with open(f"datasets/{selected_model.lower()}.txt", "r") as f:
     text = f.read()
 st.markdown(text)
 
@@ -57,13 +57,13 @@ st.markdown(text)
 st.title("2 - Model architecture")
 st.markdown("Most code generation models use GPT style architectures trained on code. Some use encoder-decoder architectures such as AlphaCode.")
 st.header("Model")
-selected_model2 = st.selectbox(
+selected_model = st.selectbox(
     "Select a code generation model", MODELS, key=2
 )
-with open(f"architectures/{selected_model2.lower()}.txt", "r") as f:
+with open(f"architectures/{selected_model.lower()}.txt", "r") as f:
     text = f.read()
 st.markdown(text)
-if model == "InCoder":
+if selected_model == "InCoder":
     st.image(INCODER_IMG, caption="Figure 1: InCoder training", width=700)
 
 # Model evaluation
@@ -75,7 +75,7 @@ st.markdown(intro)
 # Code generation
 st.title("4 - Code generation 💻")
 st.header("Models")
-selected_models3 = st.sidebar.multiselect(
+selected_models = st.multiselect(
     "Select code generation models to compare", MODELS, default=["CodeParrot"], key=3
 )
 st.header("Examples")
@@ -117,7 +117,7 @@ if st.button("Generate code!"):
             temperature=temperature,
             seed=seed,
         )
-        output = pool.map(generate_parallel, selected_models3)
+        output = pool.map(generate_parallel, selected_models)
         for i in range(len(output)):
-            st.markdown(f"**{selected_models3[i]}**")
+            st.markdown(f"**{selected_models[i]}**")
             st.code(output[i])