xzyao commited on
Commit
2ab9fff
1 Parent(s): 5e48162

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -12,12 +12,12 @@ def infer(
12
  prompt,
13
  model_name,
14
  max_new_tokens=10,
15
- temperature=0.0,
16
  top_p=1.0,
17
  top_k=40,
18
  num_completions=1,
19
  seed=42,
20
- stop="\n"
21
  ):
22
  model_name_map = {
23
  "GPT-JT-6B-v1": "Together-gpt-JT-6B-v1",
@@ -120,7 +120,7 @@ def main():
120
  st.session_state.prompt = "Please answer the following question:\n\nQuestion: In which country is Zurich located?\nAnswer:"
121
 
122
  if 'temperature' not in st.session_state:
123
- st.session_state.temperature = "0.0"
124
 
125
  if 'top_p' not in st.session_state:
126
  st.session_state.top_p = "1.0"
@@ -147,7 +147,7 @@ def main():
147
  top_p = st.text_input('top_p', st.session_state.top_p)
148
  # num_completions = st.text_input('num_completions (only the best one will be returend)', "1")
149
  num_completions = "1"
150
- stop = st.text_input('stop, split by;', r'\n')
151
  # seed = st.text_input('seed', "42")
152
  seed = "42"
153
 
 
12
  prompt,
13
  model_name,
14
  max_new_tokens=10,
15
+ temperature=0.1,
16
  top_p=1.0,
17
  top_k=40,
18
  num_completions=1,
19
  seed=42,
20
+ stop="."
21
  ):
22
  model_name_map = {
23
  "GPT-JT-6B-v1": "Together-gpt-JT-6B-v1",
 
120
  st.session_state.prompt = "Please answer the following question:\n\nQuestion: In which country is Zurich located?\nAnswer:"
121
 
122
  if 'temperature' not in st.session_state:
123
+ st.session_state.temperature = "0.1"
124
 
125
  if 'top_p' not in st.session_state:
126
  st.session_state.top_p = "1.0"
 
147
  top_p = st.text_input('top_p', st.session_state.top_p)
148
  # num_completions = st.text_input('num_completions (only the best one will be returend)', "1")
149
  num_completions = "1"
150
+ stop = st.text_input('stop, split by;', r'.')
151
  # seed = st.text_input('seed', "42")
152
  seed = "42"
153