Update app.py

app.py

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