lvwerra HF staff commited on
Commit
4b85d9d
1 Parent(s): 87be16a
Files changed (1) hide show
  1. app.py +2 -1
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 🦜")