Spaces:
Running
Running
Asankhaya Sharma
commited on
Commit
·
3964130
1
Parent(s):
a3e25d7
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
|
36 |
|
37 |
#prompts
|
38 |
-
st.title("Write with
|
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.
|
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 |
|