xzyao commited on
Commit
af56d2c
1 Parent(s): f203ba6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ from ast import literal_eval
7
  def infer(prompt,
8
  model_name,
9
  max_new_tokens=10,
10
- temperature=0.0,
11
  top_p=1.0,
12
  num_completions=1,
13
  seed=42,
@@ -36,7 +36,7 @@ col1, col2 = st.columns([1, 3])
36
  with col1:
37
  model_name = st.selectbox("Model", ["GPT-JT-6B-v1"])
38
  max_new_tokens = st.text_input('Max new tokens', "10")
39
- temperature = st.text_input('temperature', "0.0")
40
  top_p = st.text_input('top_p', "1.0")
41
  num_completions = st.text_input('num_completions (only the best one will be returend)', "1")
42
  stop = st.text_input('stop, split by;', r'\n')
 
7
  def infer(prompt,
8
  model_name,
9
  max_new_tokens=10,
10
+ temperature=0.8,
11
  top_p=1.0,
12
  num_completions=1,
13
  seed=42,
 
36
  with col1:
37
  model_name = st.selectbox("Model", ["GPT-JT-6B-v1"])
38
  max_new_tokens = st.text_input('Max new tokens', "10")
39
+ temperature = st.text_input('temperature', "0.8")
40
  top_p = st.text_input('top_p', "1.0")
41
  num_completions = st.text_input('num_completions (only the best one will be returend)', "1")
42
  stop = st.text_input('stop, split by;', r'\n')