juewang commited on
Commit
157e1ad
1 Parent(s): 5c2bbf9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -56,7 +56,7 @@ col1, col2 = st.columns([1, 3])
56
  with col1:
57
  model_name = st.selectbox("Model", ["GPT-JT-6B-v1"])
58
  max_new_tokens = st.text_input('Max new tokens', "10")
59
- temperature = st.text_input('temperature', "1.0")
60
  top_p = st.text_input('top_p', "1.0")
61
  num_completions = st.text_input('num_completions', "1")
62
  seed = st.text_input('seed', "42")
@@ -77,6 +77,7 @@ with col2:
77
 
78
  if button_submit:
79
  with st.spinner(text="In progress.."):
 
80
  report_text = infer(
81
  prompt, model_name=model_name, max_new_tokens=max_new_tokens, temperature=temperature, top_p=top_p,
82
  num_completions=num_completions, seed=seed,
 
56
  with col1:
57
  model_name = st.selectbox("Model", ["GPT-JT-6B-v1"])
58
  max_new_tokens = st.text_input('Max new tokens', "10")
59
+ temperature = st.text_input('temperature', "0.0")
60
  top_p = st.text_input('top_p', "1.0")
61
  num_completions = st.text_input('num_completions', "1")
62
  seed = st.text_input('seed', "42")
 
77
 
78
  if button_submit:
79
  with st.spinner(text="In progress.."):
80
+ generated_area.markdown("...")
81
  report_text = infer(
82
  prompt, model_name=model_name, max_new_tokens=max_new_tokens, temperature=temperature, top_p=top_p,
83
  num_completions=num_completions, seed=seed,