Spaces:

codeparrot
/

codeparrot-generation

Build error

lvwerra HF Staff commited on Mar 15, 2022

Commit

e84ce54

1 Parent(s): e804387

use small model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ def load_examples():
 st.set_page_config(page_icon=':parrot:', layout="wide")
-model_ckpt = "lvwerra/codeparrot"
 tokenizer = load_tokenizer(model_ckpt)
 model = load_model(model_ckpt)
 pipe = pipeline(model=model, tokenizer=tokenizer)

 st.set_page_config(page_icon=':parrot:', layout="wide")
+model_ckpt = "lvwerra/codeparrot-small"
 tokenizer = load_tokenizer(model_ckpt)
 model = load_model(model_ckpt)
 pipe = pipeline(model=model, tokenizer=tokenizer)