Update app.py
Browse files
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 = "
|
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"
|
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)
|