Stefan Dumitrescu commited on
Commit
50f4f41
1 Parent(s): 9194a02
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -19,10 +19,19 @@ model_list = [
19
  ]
20
 
21
  st.sidebar.header("Select model")
22
- model_checkpoint = st.sidebar.radio("", model_list)
 
 
 
 
 
 
 
 
 
23
 
24
  st.sidebar.header("Select generation parameters")
25
- max_length = st.sidebar.slider("Max Length", value=20, min_value=10, max_value=200)
26
  temperature = st.sidebar.slider("Temperature", value=1.0, min_value=0.0, max_value=1.0, step=0.05)
27
  top_k = st.sidebar.slider("Top-k", min_value=0, max_value=15, step=1, value=0)
28
  top_p = st.sidebar.slider("Top-p", min_value=0.0, max_value=1.0, step=0.05, value=0.9)
 
19
  ]
20
 
21
  st.sidebar.header("Select model")
22
+ model_checkpoint = st.sidebar.selectbox("Select model", model_list)
23
+
24
+
25
+ t1, t2, t3, t4, t5, t6 = st.tabs(["Greedy", "Beam-search", "Sampling", "Top-k Sampling", "Top-p Sampling", "Typical Sampling"])
26
+ t1.write("Greedy")
27
+ t2.write("Greedy2")
28
+ t3.write("Greedy3")
29
+ t4.write("Greedy4")
30
+
31
+
32
 
33
  st.sidebar.header("Select generation parameters")
34
+ max_length = t5.slider("Max Length", value=20, min_value=10, max_value=200)
35
  temperature = st.sidebar.slider("Temperature", value=1.0, min_value=0.0, max_value=1.0, step=0.05)
36
  top_k = st.sidebar.slider("Top-k", min_value=0, max_value=15, step=1, value=0)
37
  top_p = st.sidebar.slider("Top-p", min_value=0.0, max_value=1.0, step=0.05, value=0.9)