Spaces:
Build error
Build error
fix seed
Browse files
app.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
import streamlit as st
|
2 |
-
from transformers import AutoTokenizer, AutoModelForCausalLM
|
3 |
from transformers import pipeline
|
4 |
|
5 |
@st.cache(allow_output_mutation=True)
|
@@ -20,6 +20,7 @@ def print_hello_world():\
|
|
20 |
model_ckpt = "lvwerra/codeparrot"
|
21 |
tokenizer = load_tokenizer(model_ckpt)
|
22 |
model = load_model(model_ckpt)
|
|
|
23 |
gen_kwargs = {}
|
24 |
|
25 |
st.title("CodeParrot 🦜")
|
|
|
1 |
import streamlit as st
|
2 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed
|
3 |
from transformers import pipeline
|
4 |
|
5 |
@st.cache(allow_output_mutation=True)
|
|
|
20 |
model_ckpt = "lvwerra/codeparrot"
|
21 |
tokenizer = load_tokenizer(model_ckpt)
|
22 |
model = load_model(model_ckpt)
|
23 |
+
set_seed(42)
|
24 |
gen_kwargs = {}
|
25 |
|
26 |
st.title("CodeParrot 🦜")
|