Asankhaya Sharma commited on
Commit
3964130
·
1 Parent(s): a3e25d7
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -32,16 +32,16 @@ def infer(input_ids, bad_words_ids, max_tokens, temperature, top_k, top_p):
32
 
33
  return output_sequences
34
 
35
- default_value = "We are building the world's first "
36
 
37
  #prompts
38
- st.title("Write with a VC 🦄")
39
  st.write("This is a LLM that was fine-tuned on a dataset of investment memos to help you generate your next pitch.")
40
 
41
  sent = st.text_area("Text", default_value)
42
 
43
  max_tokens = st.sidebar.slider("Max Tokens", min_value = 16, max_value=64)
44
- temperature = st.sidebar.slider("Temperature", value = 0.8, min_value = 0.0, max_value=1.0, step=0.05)
45
  top_k = st.sidebar.slider("Top-k", min_value = 0, max_value=5, value = 4)
46
  top_p = st.sidebar.slider("Top-p", min_value = 0.0, max_value=1.0, step = 0.05, value = 0.9)
47
 
 
32
 
33
  return output_sequences
34
 
35
+ default_value = "We are building the first ever"
36
 
37
  #prompts
38
+ st.title("Write with vcGPT 🦄")
39
  st.write("This is a LLM that was fine-tuned on a dataset of investment memos to help you generate your next pitch.")
40
 
41
  sent = st.text_area("Text", default_value)
42
 
43
  max_tokens = st.sidebar.slider("Max Tokens", min_value = 16, max_value=64)
44
+ temperature = st.sidebar.slider("Temperature", value = 0.8, min_value = 0.05, max_value=1.0, step=0.05)
45
  top_k = st.sidebar.slider("Top-k", min_value = 0, max_value=5, value = 4)
46
  top_p = st.sidebar.slider("Top-p", min_value = 0.0, max_value=1.0, step = 0.05, value = 0.9)
47