Update app.py

app.py

@@ -5,12 +5,17 @@ from tf_coder.value_search import value_search_settings
 import io
 from contextlib import redirect_stdout
 
-inputs = st.text_area('The input tensor(s) specified as key-value pairs', placeholder='''
-{'rows': [10, 20, 30],
-'cols': [1,2,3,4],}
-''')
+inputs = st.text_area('The input tensor(s) specified as key-value pairs', placeholder="{'rows': [10, 20, 30],'cols': [1,2,3,4]}")
   # The single desired output tensor.
 
+st.sidebar.header("Generation settings:")
+gen_kwargs["do_sample"] = st.sidebar.radio("Decoding strategy",  ["Require All", "Require One"]) == "Require All"
+gen_kwargs["max_new_tokens"] = st.sidebar.slider("Number of tokens to generate", value=default_length, min_value=8, step=8, max_value=256)
+if gen_kwargs["do_sample"]:
+    gen_kwargs["temperature"] = st.sidebar.slider("Temperature", value = 0.2, min_value = 0.0, max_value=2.0, step=0.05)
+    gen_kwargs["top_k"] = st.sidebar.slider("Top-k", min_value = 0, max_value=100, value = 0)
+    gen_kwargs["top_p"] = st.sidebar.slider("Top-p", min_value = 0.0, max_value=1.0, step = 0.01, value = 0.95)
+    
 settings = value_search_settings.from_dict({
       'timeout': 300,
       'only_minimal_solutions': False,