Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -27,11 +27,11 @@ if gen_mode == "beam_search":
|
|
27 |
"num_beams": num_beams,
|
28 |
"early_stopping": early_stopping
|
29 |
}
|
30 |
-
|
31 |
do_sample = True
|
32 |
-
top_k = st.sidebar.number_input("
|
33 |
-
top_p = st.sidebar.number_input("
|
34 |
-
temperature = st.sidebar.number_input("
|
35 |
gen_params = {
|
36 |
"do_sample": do_sample,
|
37 |
"top_k": top_k,
|
|
|
27 |
"num_beams": num_beams,
|
28 |
"early_stopping": early_stopping
|
29 |
}
|
30 |
+
elif gen_mode == "sampling":
|
31 |
do_sample = True
|
32 |
+
top_k = st.sidebar.number_input("top_k", value=30, min_value=0)
|
33 |
+
top_p = st.sidebar.number_input("top_p", value=0, min_value=0)
|
34 |
+
temperature = st.sidebar.number_input("temperature", value=0.7, min_value=0.0)
|
35 |
gen_params = {
|
36 |
"do_sample": do_sample,
|
37 |
"top_k": top_k,
|