joehare commited on
Commit
379be48
·
1 Parent(s): bc1df36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,9 +30,9 @@ default_value = "Once upon a time, in a galaxy far, far away...."
30
  st.title("Text completion with GPT-2")
31
 
32
  sent = st.text_area("Text", default_value, height = 275)
33
- max_length = st.sidebar.slider("Max Length", min_value = 10, max_value=30)
34
  temperature = st.sidebar.slider("Temperature", value = 1.0, min_value = 0.0, max_value=1.0, step=0.05)
35
- top_k = st.sidebar.slider("Top-k", min_value = 0, max_value=5, value = 0)
36
  top_p = st.sidebar.slider("Top-p", min_value = 0.0, max_value=1.0, step = 0.05, value = 0.9)
37
 
38
  encoded_prompt = tokenizer.encode(sent, add_special_tokens=False, return_tensors="pt")
 
30
  st.title("Text completion with GPT-2")
31
 
32
  sent = st.text_area("Text", default_value, height = 275)
33
+ max_length = st.sidebar.slider("Max Length", value=100, min_value = 10, max_value=300)
34
  temperature = st.sidebar.slider("Temperature", value = 1.0, min_value = 0.0, max_value=1.0, step=0.05)
35
+ top_k = st.sidebar.slider("Top-k", min_value = 0, max_value=5, value = 2)
36
  top_p = st.sidebar.slider("Top-p", min_value = 0.0, max_value=1.0, step = 0.05, value = 0.9)
37
 
38
  encoded_prompt = tokenizer.encode(sent, add_special_tokens=False, return_tensors="pt")