loubnabnl HF staff commited on
Commit
cddb272
1 Parent(s): 78138fb

add codegen

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import streamlit as st
8
 
9
  GITHUB_CODE = "https://huggingface.co/datasets/lvwerra/github-code"
10
  MODELS = ["CodeParrot", "InCoder", "CodeGen", "PolyCoder"]
11
- GENERATION_MODELS = ["CodeParrot", "InCoder"]
12
 
13
  @st.cache()
14
  def load_examples():
 
8
 
9
  GITHUB_CODE = "https://huggingface.co/datasets/lvwerra/github-code"
10
  MODELS = ["CodeParrot", "InCoder", "CodeGen", "PolyCoder"]
11
+ GENERATION_MODELS = ["CodeParrot", "InCoder", "CodeGen"]
12
 
13
  @st.cache()
14
  def load_examples():