Hugging Face's logo

Spaces:
codeparrot
/
codeparrot-highlighting
Running

App Files Community
loubnabnl HF staff commited on
Commit
570016e
•
1 Parent(s): 318db7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -107,7 +107,7 @@ class Transformer(nn.Module):
107
  st.set_page_config(page_icon=':parrot:', layout="wide")
108
 
109
  np.random.seed(42)
110
- model_ckpt = "codeparrot/codeparrot"
111
  tokenizer = load_tokenizer(model_ckpt)
112
  model = load_model(model_ckpt)
113
  st.markdown("<h1 style='text-align: center;'>CodeParrot 🦜</h1>", unsafe_allow_html=True)
 
107
  st.set_page_config(page_icon=':parrot:', layout="wide")
108
 
109
  np.random.seed(42)
110
+ model_ckpt = "codeparrot/codeparrot-small"
111
  tokenizer = load_tokenizer(model_ckpt)
112
  model = load_model(model_ckpt)
113
  st.markdown("<h1 style='text-align: center;'>CodeParrot 🦜</h1>", unsafe_allow_html=True)