Stefan Dumitrescu commited on
Commit
831de4c
1 Parent(s): 9d077f4
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +7 -18
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 😻
4
  colorFrom: indigo
5
  colorTo: blue
6
  sdk: streamlit
7
- sdk_version: 1.12.0
8
  app_file: app.py
9
  pinned: false
10
  ---
 
4
  colorFrom: indigo
5
  colorTo: blue
6
  sdk: streamlit
7
+ sdk_version: 1.10.0
8
  app_file: app.py
9
  pinned: false
10
  ---
app.py CHANGED
@@ -83,27 +83,16 @@ with col1:
83
 
84
  model_checkpoint = st.selectbox("Select model", model_list)
85
 
86
- st.markdown("**Step 2: Adjust specific text generation parameters**")
87
 
88
- tab_greedy, tab_sampling, tab_typical = st.tabs(["Greedy", "Sampling", "Typical Sampling"])
89
-
90
- with tab_greedy:
91
- st.caption("Greedy decoding does not have any special parameters.")
92
-
93
- with tab_sampling:
94
- top_p = st.slider("Top-p", min_value=0.0, max_value=1.0, step=0.05, value=0.9)
95
- top_k = st.slider("Top-k", min_value=0, max_value=100, step=10, value=0)
96
-
97
- with tab_typical:
98
- typical_p = st.slider("Typical-p", min_value=0., max_value=1., step=.10, value=1.0)
99
-
100
- st.markdown("""---""")
101
 
102
- st.markdown("**Step 3: Adjust common text generation parameters**")
103
 
104
- no_repeat_ngrams = st.slider("No repeat n-grams", value=2, min_value=0, max_value=3)
105
- temperature = st.slider("Temperature", value=1.0, min_value=0.1, max_value=1.0, step=0.1)
106
- max_length = st.slider("Number of tokens to generate", value=50, min_value=10, max_value=256)
107
 
108
  # st.markdown("**Step 4: Select a prompt or input your own text, and click generate in the left panel**")
109
 
 
83
 
84
  model_checkpoint = st.selectbox("Select model", model_list)
85
 
86
+ st.markdown("**Step 2: Adjust text generation parameters**")
87
 
88
+ max_length = st.slider("Number of tokens to generate", value=50, min_value=10, max_value=256)
89
+ top_k = col1.slider("Top-k", min_value=0, max_value=100, step=10, value=0)
90
+ top_p = col1.slider("Top-p", min_value=0.0, max_value=1.0, step=0.05, value=0.9)
91
+ typical_p = col1.slider("Typical-p", min_value=0., max_value=1., step=.10, value=1.0)
92
+ temperature = st.slider("Temperature", value=1.0, min_value=0.1, max_value=1.0, step=0.1)
93
+ no_repeat_ngrams = st.slider("No repeat n-grams", value=2, min_value=0, max_value=3)
 
 
 
 
 
 
 
94
 
 
95
 
 
 
 
96
 
97
  # st.markdown("**Step 4: Select a prompt or input your own text, and click generate in the left panel**")
98