update models

app.py

@@ -10,7 +10,8 @@ GITHUB_CODE = "https://huggingface.co/datasets/lvwerra/github-code"
 INCODER_IMG = (
     "https://huggingface.co/datasets/loubnabnl/repo-images/raw/main/incoder.png"
 )
-MODELS = ["CodeParrot", "InCoder"]
+MODELS = ["CodeParrot", "InCoder", "CodeGen", "PolyCoder"]
+GENERATION_MODELS = ["CodeParrot", "InCoder"]
 
 @st.cache()
 def load_examples():
@@ -78,7 +79,7 @@ col1, col2, col3 = st.columns([7,1,6])
 with col1:
     st.markdown("**Models**")
     selected_models = st.multiselect(
-    "Select code generation models to compare:", MODELS, default=["CodeParrot"], key=3
+    "Select code generation models to compare:", GENERATION_MODELS, default=["CodeParrot"], key=3
 )
     st.markdown(" ")
     st.markdown("**Examples**")