Ubuntu commited on
Commit
df71374
1 Parent(s): 5165f6b
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -156,7 +156,7 @@ def main():
156
  if button_submit:
157
  generated_area.text(prompt)
158
  report_text = infer(
159
- st.session_state.prompt, model_name=model_name, max_new_tokens=max_new_tokens, temperature=temperature, top_p=top_p, top_k=top_k,
160
  num_completions=num_completions, seed=seed, stop=literal_eval("'''"+stop+"'''"),
161
  )
162
  generated_area.text(prompt + report_text)
 
156
  if button_submit:
157
  generated_area.text(prompt)
158
  report_text = infer(
159
+ prompt, model_name=model_name, max_new_tokens=max_new_tokens, temperature=temperature, top_p=top_p, top_k=top_k,
160
  num_completions=num_completions, seed=seed, stop=literal_eval("'''"+stop+"'''"),
161
  )
162
  generated_area.text(prompt + report_text)