loubnabnl HF staff commited on
Commit
0f81a78
1 Parent(s): d5f0a3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ def load_examples():
22
 
23
  st.set_page_config(page_icon=':parrot:', layout="wide")
24
 
25
- model_ckpt = "lvwerra/codeparrot"
26
  tokenizer = load_tokenizer(model_ckpt)
27
  model = load_model(model_ckpt)
28
  pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
22
 
23
  st.set_page_config(page_icon=':parrot:', layout="wide")
24
 
25
+ model_ckpt = "codeparrot/codeparrot"
26
  tokenizer = load_tokenizer(model_ckpt)
27
  model = load_model(model_ckpt)
28
  pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)