Update app.py

app.py

@@ -20,7 +20,7 @@ st.set_page_config(page_icon=':parrot:', layout="wide")
 
 device = 0 if torch.cuda.is_available() else -1
 device_name = "GPU" if torch.cuda.is_available() else "CPU"
-model_ckpt = "lvwerra/codeparrot-small"
+model_ckpt = "lvwerra/codeparrot"
 pipe = load_pipeline(model_ckpt, device)
 examples = load_examples()
 example_names = [example["name"] for example in examples]