juewang commited on
Commit
8ff5e07
1 Parent(s): e532db6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -62,7 +62,7 @@ with col1:
62
  temperature = st.text_input('temperature', "0.0")
63
  top_p = st.text_input('top_p', "1.0")
64
  num_completions = st.text_input('num_completions (only the best one will be returend)', "1")
65
- stop = st.text_input('stop, split by;', repr('\n'))
66
  seed = st.text_input('seed', "42")
67
 
68
  with col2:
@@ -83,6 +83,6 @@ with col2:
83
  generated_area.markdown(prompt)
84
  report_text = infer(
85
  prompt, model_name=model_name, max_new_tokens=max_new_tokens, temperature=temperature, top_p=top_p,
86
- num_completions=num_completions, seed=seed, stop=literal_eval(stop),
87
  )
88
- generated_area.markdown(prompt + "_" + report_text + "_")
 
62
  temperature = st.text_input('temperature', "0.0")
63
  top_p = st.text_input('top_p', "1.0")
64
  num_completions = st.text_input('num_completions (only the best one will be returend)', "1")
65
+ stop = st.text_input('stop, split by;', r'\n')
66
  seed = st.text_input('seed', "42")
67
 
68
  with col2:
 
83
  generated_area.markdown(prompt)
84
  report_text = infer(
85
  prompt, model_name=model_name, max_new_tokens=max_new_tokens, temperature=temperature, top_p=top_p,
86
+ num_completions=num_completions, seed=seed, stop=literal_eval("'''"+stop+"'''"),
87
  )
88
+ generated_area.markdown(prompt + ("_" + report_text + "_"))